索引數組到元素數組

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