查找numpy ndarray的某些值

Question

a = np.array([[1,2,3][4,5,6][7,8,9]])
b = [0,1]

我想找到在這種情況下具有(0,1)索引為2的a值。

我已經嘗試過a[b]但是它沒有顯示出預期的結果。

我怎么解決這個問題？

Answer 1

使用a[tuple(b)] ，如下所示：

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

In [4]: b=np.array([0,1])

In [5]: a[tuple(b)]
Out[5]: 2

有關更多信息，請參閱https://docs.scipy.org/doc/numpy-1.16.0/user/basics.indexing.html#dealing-with-variable-numbers-of-indices-within-programs

Answer 2

您也可以使用for循環：

a = np.array( [[1,2,3],[4,5,6],[7,8,9]])
b = [0,1]

for var in b:
    a = a[var]
print(a)

產量

2

Answer 3

這些行：

a = [[1,2,3],[4,5,6],[7,8,9]]
b = [0,1]

這些是列表，而不是ndarrays。 列表不支持奇特的numpy索引。 要訪問該列表列表的第0個條目的第1個元素，請使用：

print(a[0][1])

如果您實際上想使用numpy ndarrays，請嘗試如下操作：

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
print(a[0,1])
print(a[0][1])