Python：替换 2D numpy 数组，用一维数组中的值填充索引

Question

我有anxm numpy python 阵列。 该数组的值介于 0 和 10 之间，与一维 numpy 数组的索引一致（一维数组的长度 = 11，因此为 0-10）。 此一维数组存储任意值。 我现在想用一维数组中的相应值替换我的二维数组的索引值。

我怎样才能做到这一点？

干杯诺克森

Answer 1

numpy 很简单： result = values_1d[arr_idxs_2d]

看：

>>> import numpy as np
>>> imin, imax = 0, 9

>>> values_1d = np.random.randn(imax+1)
>>> values_1d
array([-0.80355548,  1.00645905, -1.17436992, -1.52008027,  0.73544379,
       -0.72175806, -0.69933584, -0.50195895, -0.04588092,  0.86542796])

>>> arr_idxs_2d = np.random.randint(imin, imax, (4, 5))                                              

>>> arr_idxs_2d
array([[6, 7, 4, 3, 1],
       [8, 8, 3, 7, 7],
       [1, 1, 5, 3, 2],
       [6, 2, 3, 6, 8]])

>>> values_1d[arr_idxs_2d]
array([[-0.69933584, -0.50195895,  0.73544379, -1.52008027,  1.00645905],
       [-0.04588092, -0.04588092, -1.52008027, -0.50195895, -0.50195895],
       [ 1.00645905,  1.00645905, -0.72175806, -1.52008027, -1.17436992],
       [-0.69933584, -1.17436992, -1.52008027, -0.69933584, -0.04588092]])