python - 如何用数组索引一个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])

如何A使用我的i坐标数组进行索引。以下不起作用：

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]; 后者只是前者的语法糖。

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

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

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

Answer 2

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