从numpy中的网格中获取特定索引

Question

我运行以下代码：

import numpy as np
Z = np.arange(9).reshape(3,3)
print(Z[[0,1],[0,2]])

Z是：

0 1 2
3 4 5
6 7 8

我希望得到值1和2，但得到了0和5，所以我不明白这个索引是如何工作的。

感谢

Answer 1

Z[[0,1]，[0,2]]返回Z[0,0]和Z[1,2]，这就是为什么得到0和5的原因。