python - 重复值 n 次，n 在一个数组中

Question

我有一个值数组 a，其值在每个索引 idx 我想重复一定数量 b[idx]次，在另一个数组 (b) 中的相同索引 idx 处给出，如下所示:

a = numpy.array([1, 2, 3 ,4, 5])
b = numpy.array([2, 3, 1, 2, 4])

期望的输出:

c = numpy.array([1, 1, 2, 2, 2, 3, 4, 4, 5, 5, 5, 5])

我意识到我可以做这样的事情:

len_a = numpy.shape(a)[0]
sum_b = sum(b)
c = numpy.zeros((1, 0))
for idx in range(len_a):
    repeated_a = numpy.repeat(a[idx], b[idx])
    repeated_a = numpy.reshape(repeated_a, (1, numpy.shape(repeated_a)[0]))
    c = numpy.hstack((c, repeated_a))

但是，循环不是一个好的选择，因为它很慢。我该怎么做才能让它更快？也许是某种形式的矢量化。

Answer 1

您正在寻找为此目的而制作的内置重复功能。只需将两个数组都输入函数即可:

np.repeat(a,b)
#[1 1 2 2 2 3 4 4 5 5 5 5]