python - Numpy 索引数组到元素数组

Question

我有一个包含整数的 MxN numpy 数组，表示大小为 K 的大数组的索引如何有效地将我的 M*N 索引数组转换为 MxN 元素数组？

示例:

K = ['a','b','c','d']
M = [[0,3],[2,1]]

结果:

[['a','d'],['c','b']]

谢谢!

Answer 1

我们可以从这些列表中创建 numpy 数组:

import numpy as np

k = np.array(K)
m = np.array(M)

然后使用 k[m] 执行映射:

>>> k[m]
array([['a', 'd'],
       ['c', 'b']], dtype='<U1')

此处为 m 中的每个元素，因此我们用存储在 k 中的元素“替换”它在索引的原始值m在那个位置。