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

Python  /  管理员 发布于 5年前   784

如下所示：

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 实现对特定行或列取值就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。