取numpy数组的某一行某一列

时间:2021-04-24 15:03:52

这个操作在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]]
这才是我们想要的结果。必须要经过这两步才能完成