例如,我有一个ndarray
,它是:
a = np.array([1, 3, 5, 7, 2, 4, 6, 8])
现在我想把a
分成两部分,一部分是全部数字<;5,另一部分是全部>;=5:
[array([1,3,2,4]), array([5,7,6,8])]
当然,我可以遍历a
并创建两个新数组。但我想知道纽比能提供更好的方法吗?
类似地,对于多维数组,例如
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[2, 4, 7]])
我要根据第一列<;3和>;=3拆分它,结果是:
[array([[1, 2, 3],
[2, 4, 7]]),
array([[4, 5, 6],
[7, 8, 9]])]
有没有更好的方法来代替它?谢谢。
这将产生以下输出:
这可能是一个快速的解决方案
相关问题 更多 >
编程相关推荐