使用NumPy中的频率数组检索数组元素

YXD*_*YXD 2 python numpy

我有一系列数字,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] ]
Run Code Online (Sandbox Code Playgroud)

seb*_*erg 6

这究竟是什么np.arange(5).repeat([1,0,3,2,0]).