numpy.ndarray 实现对特定行或列取值

如下所示：

import numpy as np

b = [[1,2,0],
[4,5,0],
[7,8,1],
[4,0,1],
[7,11,1]
]
a=np.array([b]).reshape((5,3))
print(a)
c=[1,3,4]
# print(a[c])
d=np.nonzero(a[:, 2] == 0)
print(d)
print(a[d])

以上这篇numpy.ndarray 实现对特定行或列取值就是小编分享给大家的全部内容了，希望能给大家一个参考

您可能感兴趣的文章:

• numpy中实现ndarray数组返回符合特定条件的索引方法
• numpy ndarray 取出满足特定条件的某些行实例
• Python中Numpy ndarray的使用详解