当前位置:主页 > 软件编程 > Python代码 >

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

时间:2021-06-13 08:41:19 | 栏目:Python代码 | 点击:

如下所示:

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])
 
 

您可能感兴趣的文章:

相关文章