numpy如何在某个坐标处获得元素的值？

Question

我找到了获取numpy数组的最大值，然后获取坐标的方法，但有没有办法做相反的事情？

我的意思是，通过输入2D坐标来获取数组元素的值？

真的很感激帮助！

干杯，

大卫

Answer 1

语法为A[row_number,column_number]

第一行或第列编号为0.

以下是一个例子：

In [1]: import numpy as np

In [2]: A = np.array([[1,2,3],[4,5,6],[7,8,9]])

In [3]: A
Out[3]: 
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

In [4]: A[1,1]
Out[4]: 5

In [5]: A[2,1]
Out[5]: 8

In [6]: A[0,1]
Out[6]: 2

In [7]: A[2,2]
Out[7]: 9

你也可以使用元组作为坐标：

In [8]: coords = (0,0)

In [9]: A[coords]
Out[9]: 1

In [10]: newcoords = (2,2)

In [11]: A[newcoords]
Out[11]: 9

还有其他几种方法可以读取numpy数组的元素。您可能希望阅读此Indexing documentation