python - 将值列表分配给 numpy 数组

Question

假设有一个二维 numpy 数组，形状为 (10000, 100)，大小为 1000000。还有一个长度为 1000000 的一维列表。将列表中的所有值分配给数组的最快方法是什么？我的解决方案是:

my_array = np.zeros([10000, 100]) 
my_list = range(1000000)
length_of_list = len(my_list)

for i in range(length_of_list):                  
    my_array.flat[i] = my_list[i]

还有更好的吗？

Answer 1

我认为将列表转换为数组并重新调整它的形状将是最快的解决方案，但事实证明在我的设置中 @M4rtini 的单行代码甚至更快:

In [25]: %%timeit -n 10 
   ....: for i in range(length_of_list):                  
   ....:     my_array.flat[i] = my_list[i]
   ....: 
10 loops, best of 3: 420 ms per loop

In [26]: %timeit -n 10 my_array.flat[:] = my_list
10 loops, best of 3: 119 ms per loop

In [27]: %timeit -n 10 my_array = np.array(my_list).reshape(10000, 100)
10 loops, best of 3: 133 ms per loop