Заменить значения массива на основе вектора индексов

Question

У меня есть массив значений и массив индексов. Я хотел бы использовать индексы для генерации массива одинакового размера со значениями, соответствующими индексу первого массива:

vals = np.array([.2,.3])
ind = np.array([0,0,1,0])

ожидаемый результат:

np.array([.2,.2,.3,.2])

Answer 1

Вы можете просто индексировать свой массив значений с помощью массива индексов, поскольку они являются целыми числами (требуется для индексации):

vals[ind]

, что приводит к:

array([0.2, 0.2, 0.3, 0.2])

по желанию.

Answer 2

Не уверен, что это наилучший способ NumPy, но вы можете сделать:

[vals[i] for i in ind]

new_vals = np.array([vals[i] for i in ind])