NumPy中的花式索引

时间:2018-03-07 05:33:28

标签: python numpy

这里发生了什么?我无法得到它。

a = np.arange(12).reshape(3,4)
i = np.array( [ [0,1],[1,2] ] )
j = np.array( [ [2,1],[3,3] ] )
print(a[i,j])

输出看起来像这样

                 [[ 2,  5],
                 [ 7, 11]]

1 个答案:

答案 0 :(得分:1)

也许这种指数的重新排列会有所帮助

[0,1],  [2,1]
[1,2],  [3,3] 

(0,2)  (1,1)
(1,3)  (2,3)

它使用2个阵列中每个阵列的4个索引来挑选4个值。

a [0,2],a [1,1]等