numpy 创建一个长度为n的向量，其中包含n个条目“x”

bxgwgixi 于 2022-12-04 发布在其他

关注(0)|答案(4)|浏览(190)

范例：

n = 5
x = 3.5
Output:
array([3.5, 3.5, 3.5, 3.5, 3.5])

我的密码：

import numpy as np
def init_all_x(n, x):
    np.all = [x]*n
    return np.all
init_all_x(5, 3.5)

我的问题：

为什么init_all_x（5，3.5）.形状无法运行？
如果我的代码是错误的，正确的代码是什么？谢谢！

4条答案

可以使用np.ones

arr = np.ones(5)*3.5

使用numpy.repeat的简单方法：

n = 5
x = 3.5
a = np.repeat(x, n)

输出量：

array([3.5, 3.5, 3.5, 3.5, 3.5])

对于您的要求，不需要使用Numpy lib，您可以这样编写代码：

def init_all_x(n, x):
    return [x]*n

p = init_all_x(5, 3.5)
print(p)

输出量：

[3.5, 3.5, 3.5, 3.5, 3.5]

Numpy有一个专门的函数np.full来完成这个任务：

n = 5
x = 3.5

out = np.full(n, x)
# array([3.5, 3.5, 3.5, 3.5, 3.5])