py: numpy提取某些列

https://www.runoob.com/numpy/numpy-ndexing-and-slicing.html

切片還可以包括省略號 …，來使選擇元組的長度與數組的維度相同。 如果在行位置使用省略號，它將返回包含行中元素的 ndarray。

import numpy as np
 
a = np.array([[1,2,3],[3,4,5],[4,5,6]])  
print (a[...,1])   # 第2列元素
print (a[1,...])   # 第2行元素
print (a[...,1:])  # 第2列及剩下的所有元素
# 輸出結果爲：

[2 4 5]
[3 4 5]
[[2 3]
 [4 5]
 [5 6]]