Numpy：从一维整数数组创建新数组，其中整数数组中的每个元素表示新数组中的元素数

Question

这看起来应该很简单，但我很难过（对 numpy 来说也很新。）

我有一个一维整数数组a 。

我想生成一个新的一维数组b使得：

• b中的元素的数量等于在所述元素的总和
• 在B中的值等于由相应的元件以分开的一些任意的常数。

这是一口，所以这里有一个我想让它更具体的例子：

a = array([2,3,3,4])
CONSTANT = 120
.
.
.
b = array([60,60,
           40,40,40,
           40,40,40,
           30,30,30,30])

任何帮助表示赞赏！

Answer 1

我认为一个非常明确的方法是

import numpy as np
a = np.array([2,3,3,4]) 
constant = 120

#numpy.repeat(x,t) repeats the val x t times you can use x and t as vectors of same len
b = np.repeat(constant/a , a)

Answer 2

你可以用np.concatenate和老式的zip来做到这一点：

>>> elements = CONSTANT / a
>>> np.concatenate([np.array([e]*n) for e, n in zip(elements, a)])
array([60., 60., 40., 40., 40., 40., 40., 40., 30., 30., 30., 30.])