索引数组到元素数组

Question

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

示例：

Parse(const unsigned char * pStart, unsigned short nLen)

结果：

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

谢谢！

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原始值的索引处。