這個操作在numpy數組上的操作感覺有點麻煩,但是也沒辦法。
例如
a = [[1,2,3],
[4,5,6],
[7,8,9]]
取 a 的 2 3 行, 1 2 列
c=[1,2]
d =[0,1]
若寫爲
b = a[c,d]
output:
[4 8]
取的是 第二行第一列 和第三行第二列的數據
這並不是我們想要的結果。
正確做法是:
b = a[c]先取想要的行數據
b = b[:,d]
print(b)
output:
[[4 5]
[7 8]]
這纔是我們想要的結果。必須要經過這兩步才能完成