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

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)]))

它给出了期望的结果，但是有没有更简单的方法来达到期望的输出呢？