如何使用数组索引numpy数组元素

Question

我有一个numpy数组，想获取特定元素的值。 例如，我可能想访问[1,1]处的值

import numpy as np
A = np.arange(9).reshape(3,3)
print A[1,1]
# 4

现在，说我将坐标放在数组中：

i = np.array([1,1])

我如何用我的i坐标数组索引A 以下内容不起作用：

print A[i]
# [[3 4 5]
#  [3 4 5]]

Answer 1

• http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html

在Python中， x[(exp1, exp2, ..., expN)]等效于x[exp1, exp2, ..., expN] ; 后者只是前者的语法糖。

因此，要获得与A[1,1]相同的结果，您必须使用元组进行索引。

如果使用ndarray作为索引对象，则会触发高级索引：

• http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#advanced-indexing

Answer 2

最好的选择是A[tuple(i)] 。 tuple(i)调用只是将i视为一个序列，并将序列项放入一个元组中。 请注意，如果您的数组有多个维度，则不会构成嵌套元组。 不过，在这种情况下，这并不重要。