如何使用另一个numpy数组索引多维numpy数组

时间:2017-06-17 11:18:00

标签: python numpy

我们假设我有以下numpy数组:

idx = [1,2]
A   = [[1,2,3],
       [4,5,6],
       [7,8,9]]

我想获得A[idx[0],idx[1]]

A[idx]获得一分。出于清晰原因,我不想使用A[idx[0],idx[1]]

1 个答案:

答案 0 :(得分:2)

我找到了一个解决方案:

A[tuple(indx)]