Numpy：将索引数组转换为元素数组答案

【问题标题】：Numpy: Turning an index array into an element arrayNumpy：将索引数组转换为元素数组
【发布时间】：2022-01-02 11:21:30
【问题描述】：

假设我有一个索引数组：

A = [1, 0, 3, 2, 0, 1]

对于另一个值数组：

B = [21, 33, 10, 2]

那我想要一个新数组：

C = [B[A[0]], B[[A[1]], ..., B[A[[5]]] = [B[1], B[0], ..., B[1]] = [33, 21, ..., 21]

如何使用 numpy 数组做到这一点？

【问题讨论】：

【解决方案1】：

当B 是一个 numpy 数组时，您可以应用 numpy 索引：

import numpy as np

A = [1, 0, 3, 2, 0, 1]
B = np.array([21, 33, 10, 2])
output = B[A]
print(output) # [33 21  2 10 21 33]

【讨论】：