python - 根据另一个索引数组重复 numpy 数组元素

Question

我有以下占位符数组:

placeholder = np.zeros(10)

A 包含我想要进入占位符的值

A = np.array([25, 40, 65,50])

idx 具有占位符的索引(请注意，这与 A 具有相同的形状)

idx = np.array([0, 5, 6, 8])

问题:

我想用 A 的元素填充 placeholder。 A 中元素的重复次数由 idx 数组的间隔长度定义。

比如25重复5次，因为对应的索引范围是[0, 5)。 65重复了两次，因为对应的索引范围是[6,8)

预期输出:

np.array([25, 25, 25, 25, 25, 40, 65, 65, 50, 50])

Answer 1

使用 np.diff + np.repeat 快速简单:

repeats = np.diff(np.append(idx, len(placeholder)))

A.repeat(repeats)
array([25, 25, 25, 25, 25, 40, 65, 65, 50, 50])