python

超轻量级php框架startmvc

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

更新时间:2020-08-12 15:18:01 作者:startmvc
如下所示:importnumpyasnpb=[[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

如下所示:


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 取值