如何理解这个python代码，非常感谢

Question

import numpy as np
p = np.array([[1,2,3]])
print(p[np.array([0]), np.array([1,0,0])])

# output:[2,1,1]

我试图理解为什么输出此结果。

Answer 1

p是（1,3）形状数组。索引，也可以写为

p[ 0, [1,0,0]]

选择p[0,1]，p[0,0]和p[0,0]，即2和1（两次）。

使用列表或数组进行直接索引，也称为高级索引。