numpy：将频率阵列分配

Question

在带有numpy的python中，最快的方法是将数组变成

array([0,2,3,1,0,0,1])

放入另一个数组

array([1,1,2,2,2,3,6])

其中第一个数组给出每个索引的频率（即，索引0的频率为0，索引1的频率为2，索引2的频率为3，依此类推），第二个数组重复每个索引的次数为在第一个数组中指定。

背景：在给定长度为k频率矢量f我用它来“填充”（我找不到更好的词）一个k × k矩阵M （稀疏与否）：

f  = np.array([0,2,3,1,0,0,1])
f_ = np.array([1,1,2,2,2,3,6])
M_ = M[f_[:,None],f_]

Answer 1

在范围数组上使用np.repeat覆盖输入数组的长度以及数组本身的重复次数-

np.repeat(np.arange(len(a)), a)

样品运行-

In [109]: a
Out[109]: array([0, 2, 3, 1, 0, 0, 1])

In [110]: np.repeat(np.arange(len(a)), a)
Out[110]: array([1, 1, 2, 2, 2, 3, 6])