在指数

Question

使用NumPy创建一个新列表的方法是什么？该列表包含另一个列表中所需元素的子集？

indices = [0,2,1,1]
a = [8,7,6,5,4]
b = [a[i] for i in indices] # returns [8,6,7,7]

如果可能，我正在寻找单一功能。像这样：

import numpy as np
indices = np.array([0,2,1,1])
a = np.array([8,7,6,5,4])
b = np.get_selected_elements(a, indices)