python - 在 Python 中索引的其他形式

Question

我正在学习机器学习类(class)，在对 numpy 库的评论中，使用了一种我以前从未见过的索引方法。我们定义了 a = np.array([[1,2],[3,4],[5,6]]) .随后创建了两个新数组:np.array([a[0,0], a[1,1], a[2,1]] .这是我所知道的索引方式。我不知道的索引方式是这样的:print(a[[0,1,2],[0,1,1]]) .有人可以帮我解释一下这种最新形式的索引吗？

Answer 1

当n传递数组，其中 n是数组维数，它们被用作对应维数的索引。 a[[0,1,2],[0,1,1]]相当于 [a[0,0], a[1,1], a[2,1]]让我引述 numpy文档:

... From each row, a specific element should be selected. The rowindex is just [0, 1, 2] and the column index specifies the element tochoose for the corresponding row, here [0, 1, 0]. Using both togetherthe task can be solved using advanced indexing:

x = np.array([[1, 2], [3, 4], [5, 6]])

x[[0, 1, 2], [0, 1, 0]]
array([1, 4, 5])