python - NumPy 数组初始化(填充相同的值)

Question

我需要创建一个长度为n的NumPy数组，每个元素都是v。

还有什么比:

a = empty(n)
for i in range(n):
    a[i] = v

我知道 zeros 和 ones 适用于 v = 0, 1。我可以使用 v * ones(n)，但它当 v 为 None 时，不起作用，而且会慢很多。

Answer 1

引入了 NumPy 1.8 np.full() ，这是一种比 empty() 后跟 fill() 更直接的方法，用于创建一个填充了某个值的数组:

>>> np.full((3, 5), 7)
array([[ 7.,  7.,  7.,  7.,  7.],
       [ 7.,  7.,  7.,  7.,  7.],
       [ 7.,  7.,  7.,  7.,  7.]])

>>> np.full((3, 5), 7, dtype=int)
array([[7, 7, 7, 7, 7],
       [7, 7, 7, 7, 7],
       [7, 7, 7, 7, 7]])

这可以说是创建一个填充了特定值的数组的方式，因为它明确地描述了要实现的目标(原则上它可以非常有效，因为它执行一项非常具体的任务)。