它是如何进行索引的？为什么输出是[1,4,5]？

我正在按照http://cs231n.github.io/python-numpy-tutorial/#numpy上的教程进行学习

a = np.array([[1,2], [3, 4], [5, 6]])# 整数数组索引的示例。# 返回的数组将具有形状(3,) 和print(a[[0, 1, 2], [0, 1, 0]])  # 打印 "[1 4 5]"

回答：

这在numpy中被称为高级索引。

你可以将第一个列表和第二个列表想象成x轴和y轴。因此，a[[0,1,2],[0,1,0]]就像是从a中获取三个元素，它们的坐标分别是(0,0), (1,1), (2,0)。

a[0,0] # 1a[1,1] # 4a[2,0] # 5