python - 是否有一种单行解决方案可以将标量转换为每个元素中具有相同值的 n 长度数组

Question

我正在寻找一种替代方案

a = 3
n = 5
b = np.ones(n)
b *= a

基本上，我想创建一个大小为 n 的向量，并且每个项目都具有相同的值。

标签： pythonnumpy

Answer 1

您可以使用np.full创建一个给定大小的空数组并指定一个fill_value：

np.full(n, a)
# array([3, 3, 3, 3, 3])