NumPy分割多维数组

Question

我有一个多维numpy数组，我想根据特定的列进行拆分。

例如：[[1,0,2,3],[1,2,3,4],[2,3,4,5]]。假设我想通过表达式x <=2来拆分这个数组。那么我将得到两个数组，[[1,0,2,3],[1,2,3,4]]和[[2,3,4,5]]。

我目前正在使用这个语句，但我认为它不正确。

splits = np.split(S, np.where(S[:, a] <= t)[0][:1]) #splits S based on t

#a is the column number

- mrQWERTY

---

x<=2是什么意思？x是您的行号吗？ - orange

这只是特定列中的值。 - mrQWERTY

您IP地址为143.198.54.68，由于运营成本限制，当前对于免费用户的使用频率限制为每个IP每72小时10次对话，如需解除限制，请点击左下角设置图标按钮（手机用户先点击左上角菜单按钮）。 - orange

@orange 是的，完全正确。 - mrQWERTY

请查看我的答案。 - orange

Answer 1

>>> import numpy as np
>>> a = np.asarray([[1,0,2,3],[1,2,3,4],[2,3,4,5]])
>>> a
array([[1, 0, 2, 3],
       [1, 2, 3, 4],
       [2, 3, 4, 5]])
>>> split1 = a[a[:,1] <= 2, :]
array([[1, 0, 2, 3],
       [1, 2, 3, 4]])
>>> split2 = a[a[:,1] > 2, :]
array([[2, 3, 4, 5]])

- orange