Python使用变量索引一个numpy数组

Question

我有一个索引位置的变量作为列表，例如 [1,2,3]。

我想使用此变量从多维数组中选择该索引。所以像：

A = np.array([[0,1],[2,2]])
location = [1,1]
print(A[location])

我希望它给我 2，与这相同：

print(A[1,1])

但我只有我的索引在一个变量中。有任何想法吗？

Answer 1

这将是反高潮的，但只需使用一个元组：

>>> A[tuple(location)]
2

（或只是设置location = (1,1)）。作为索引的列表具有特殊的行为，其中矩阵为每个元素单独编制索引，结果被连接。一个特定的结果是您可以拥有比维度更多的元素，这通常也令人惊讶。