标签: python numpy vectorization
我有一个类似这样的计数的向量
[2, 4, 3]
并且我正在尝试创建其索引数组
[0, 1, 2]
每个索引重复对应计数指定的次数,即最终输出应为
[0, 0, 1, 1, 1, 1, 2, 2, 2]
使用for循环执行此操作很容易,但效率很低。有快速的矢量化解决方案吗?