numpy.take 

numpy的. take  （ a ， indices ， axis = None ， out = None ， mode ='raise' ）

沿軸取數組中的元素。

這個函數與“花式”索引（使用數組索引數組）的功能相同; 但是，如果您需要沿給定軸的元素，則可以更容易使用。

參數：	a：array_like 源數組。 indices：array_like 要提取的值的索引。 1.8.0版中的新功能。 也允許標量為指數。 axis：int，可選 用於選擇值的軸。默認情況下，使用展平的輸入數組。 out：ndarray，可選 如果提供，結果將放在此數組中。它應該是適當的形狀和dtype。 mode：{'raise'，'wrap'，'clip'}，可選 指定越界索引的行為方式。 'raise' - 引發錯誤（默認） '包裹' - 環繞 'clip' - 剪輯到范圍 'clip'模式意味着所有過大的索引都被索引沿着該軸的最后一個元素所取代。請注意，這會禁用帶負數的索引。
返回：	子陣：ndarray 返回的數組與a的類型相同。

a = np.array([[1, 2],
             [3, 4]])
b = np.take(a,[0,1])
print(b)
>>[1 2]

a = np.array([[1, 2],
             [3, 4]])
b = np.take(a,0,axis=0)
print(b)
>>[1 2]

a = np.array([[1, 2],
             [3, 4]])
b = np.take(a,1,axis=1)
print(b)
>>[2 4]

a = np.array([[1, 2],
             [3, 4]])
b = np.take(a,[0,1],axis=1)
print(b)
>>[[1,2]
　 [3,4]]

常用參數：
　　indices：參數代表你要截取數據的索引
　　axis：代表你要怎么截取，是按行截取還是按列截取，axis=1，豎着截取，axis=0橫着截取