将向量的每个元素重复由另一个计数向量提供的次数

时间:2019-11-01 18:21:43

标签: python numpy vectorization

我有一个类似这样的计数的向量

[2, 4, 3]

并且我正在尝试创建其索引数组

[0, 1, 2]

每个索引重复对应计数指定的次数,即最终输出应为

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

使用for循环执行此操作很容易,但效率很低。有快速的矢量化解决方案吗?

0 个答案:

没有答案