python - 在 NumPy 中检索具有频率数组的数组元素

Question

我有一个数字数组，a。我有第二个数组，b，指定我想检索 a 中相应元素的次数。如何实现？在这种情况下，输出的顺序并不重要。

import numpy as np

a = np.arange(5)
b = np.array([1,0,3,2,0])

# desired output = [0,2,2,2,3,3]
# i.e. [a[0], a[2], a[2], a[2], a[3], a[3] ]

Answer 1

这正是 np.arange(5).repeat([1,0,3,2,0]) 所做的。