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

还有什么比:

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

我知道0和1适用于v = 0,1。我可以用v * ones(n),但当v为None时行不通,也会慢得多。


当前回答

你也可以使用np.broadcast_to。

要创建一个形状(维度)为s,值为v的数组,你可以这样做(在你的例子中,数组是1-D,并且s = (n,)):

a = np.broadcast_to(v, s).copy()

如果a只需要是只读的,你可以使用以下方法(这是更有效的方式):

a = np.broadcast_to(v, s)

这样做的好处是v可以被赋值为单个数字,但如果需要不同的值,也可以赋值为数组(只要v.shape匹配s的尾部)。

额外的好处:如果你想强制创建的数组的dtype:

a = np.broadcast_to(np.asarray(v, dtype), s).copy()

其他回答

显然,不仅绝对速度,而且速度顺序(如user1579844所报告的)都与机器有关;以下是我的发现:

一个= np.empty (1 e4);A.fill(5)是最快的;

按速度递减排列:

timeit a=np.empty(1e4); a.fill(5) 
# 100000 loops, best of 3: 10.2 us per loop
timeit a=np.empty(1e4); a[:]=5
# 100000 loops, best of 3: 16.9 us per loop
timeit a=np.ones(1e4)*5
# 100000 loops, best of 3: 32.2 us per loop
timeit a=np.tile(5,[1e4])
# 10000 loops, best of 3: 90.9 us per loop
timeit a=np.repeat(5,(1e4))
# 10000 loops, best of 3: 98.3 us per loop
timeit a=np.array([5]*int(1e4))
# 1000 loops, best of 3: 1.69 ms per loop (slowest BY FAR!)

所以,试着找出答案,并使用你平台上最快的方法。

我相信填充是最快的方法。

a = np.empty(10)
a.fill(7)

您还应该始终避免像您在示例中所做的那样进行迭代。简单的A [:] = v将完成使用numpy广播所做的迭代。

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]])

这可以说是创建一个充满特定值的数组的方法,因为它显式地描述了要实现的内容(原则上它可以非常有效,因为它执行非常特定的任务)。

你也可以使用np.broadcast_to。

要创建一个形状(维度)为s,值为v的数组,你可以这样做(在你的例子中,数组是1-D,并且s = (n,)):

a = np.broadcast_to(v, s).copy()

如果a只需要是只读的,你可以使用以下方法(这是更有效的方式):

a = np.broadcast_to(v, s)

这样做的好处是v可以被赋值为单个数字,但如果需要不同的值,也可以赋值为数组(只要v.shape匹配s的尾部)。

额外的好处:如果你想强制创建的数组的dtype:

a = np.broadcast_to(np.asarray(v, dtype), s).copy()

为Numpy 1.7.0更新:(向@Rolf Bartstra致敬。)

一个= np.empty (n);A.fill(5)最快。

按速度递减排列:

%timeit a=np.empty(10000); a.fill(5)
100000 loops, best of 3: 5.85 us per loop

%timeit a=np.empty(10000); a[:]=5 
100000 loops, best of 3: 7.15 us per loop

%timeit a=np.ones(10000)*5
10000 loops, best of 3: 22.9 us per loop

%timeit a=np.repeat(5,(10000))
10000 loops, best of 3: 81.7 us per loop

%timeit a=np.tile(5,[10000])
10000 loops, best of 3: 82.9 us per loop