python - 给定索引应重复的次数，如何创建索引数组？

Question

给定索引应重复的次数，如何创建索引数组？

鉴于：

[ 3, 2, 5]

所需的输出是：

[0, 0, 0, 1, 1, 2, 2, 2, 2, 2]

我试过这样的事情：

from itertools import chain
import numpy as np

x = np.array([ 3, 2, 5])


list(chain(*[[i]*v for i,v in enumerate(x)]))

它给出了预期的结果，但有没有更简单的方法来实现预期的输出？

标签： pythonlistnumpyindices