1. 程式人生 > 程式設計 >numpy.ndarray 實現對特定行或列取值

numpy.ndarray 實現對特定行或列取值

如下所示:

import numpy as np
 
b = [[1,2,0],[4,5,[7,8,1],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 實現對特定行或列取值就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。