从索引数组到元素数组的Numpy数组

Question

我有一个包含int的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原始值索引处的元素来“替换”它