1. 程式人生 > >np.where和np.bincount詳解

np.where和np.bincount詳解

簡單用法

a = np.array(xrange(5*3))
a.shape =(5,3)

對這樣一個數組進行操作:
array([[0,1,2],[3,4,5],[6,7,8],[9,10,11],[12,13,14]])

  1. 尋找以9開頭的某一行:
    a[np.where(a[:,0]==9)] —->#array([[ 9, 10, 11]])

  2. 尋找l列中某一值等於4的某一行:
    a[np.where(a[:,1]==4)] —->array([[3, 4, 5]])

  3. 尋找某一行:第一位值為0,第二位值為1:
    a[np.where((a[:,0]==0)*(a[:,1]==1))]#array([[0,1, 2]])

np.bincount詳解