python – Numpy：数组的索引

概述给出以下示例 d = array([[1, 2, 3], [1, 2, 3], [1, 3, 3], [4, 4, 4], [5, 5, 5] ]) 要在第一列中获取包含1的子数组： d[ d[:,0] == 1 ]array([[1, 2, 3], [1, 2, 3], 给出以下示例

d = array([[1,2,3],[1,3,[4,4,4],[5,5,5]          ])

要在第一列中获取包含1的子数组：

d[ d[:,0] == 1 ]array([[1,3]])

如何得到(没有循环)包含1和5的子数组？不应该是这样的

d[ d[:,0] == [1,5] ]  #   ---> array([1,3])

这不起作用？

解决方法 方法#1：使用按位或|结合条件：

>>> darray([[1,5]])>>> (d[:,0] == 1) | (d[:,0] == 5)array([ True,True,False,True],dtype=bool)>>> d[(d[:,0] == 5)]array([[1,5]])

方法#2：使用np.in1d,如果有很多值,这可能更容易：

>>> np.in1d(d[:,0],5])array([ True,dtype=bool)>>> d[np.in1d(d[:,5])]array([[1,5]])

总结

以上是内存溢出为你收集整理的python – Numpy：数组的索引全部内容，希望文章能够帮你解决python – Numpy：数组的索引所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。