ndarray 的快速突变（替换 numpy ndarray 的一部分）

Question

我正在寻找最快的方法来替换部分预分配的 numpy ndarray。 下面你可以看到sim_matrix（2D ndarray）。 “For 循环”迭代它rowwise 并覆盖levels (numpy.array) 给出的值。

有快速的方法，找到了：

1. numpy.copyto()
2. numpy.fromiter()
3. = 分配给切片

注意：为简单起见，我在每次迭代中分配相同的值，但在生产设置中并非如此。在实际设置中，levels 每行包含不同的值。因此，不鼓励将快捷方式用作numpy.repeat 或numpy.hstack 作为答案。

有没有更快的方法来更改 ndarray 的某些部分？

numpy_overwrite.py：

import numpy
panel_size = 365 * 7
rows = 1 * (1+1+2*4) * 10000
# rows = 100
levels = [1778.24] * panel_size
levels_array = numpy.fromiter(levels,dtype='single', count = len(levels))
sim_matrix = numpy.empty([rows, panel_size],dtype='single')

def _copyto(sim_matrix_, level_):
    for i in range(sim_matrix_.shape[0]):
        numpy.copyto(sim_matrix[i], level_)


def _fromiter(sim_matrix_, level_):
    len_level = len(level_)
    for i in range(sim_matrix_.shape[0]):
        sim_matrix[i] = numpy.fromiter(level_,
                                      dtype='single', 
                                      count = len_level)

def _just_assign(sim_matrix_, level_):
    for i in range(sim_matrix_.shape[0]):
        sim_matrix[i] = level_

---

numpy_overwrite_test.py

import timeit
import numpy_overwrite as npo
print('_fromiter')
print(timeit.timeit('npo._fromiter(npo.sim_matrix, npo.levels)', setup="import numpy_overwrite as npo;", number=3))
print(timeit.timeit('npo._fromiter(npo.sim_matrix, npo.levels_array)', setup="import numpy_overwrite as npo;", number=3))
print('_copyto')
print(timeit.timeit('npo._copyto(npo.sim_matrix, npo.levels)', setup="import numpy_overwrite as npo;", number=3))
print(timeit.timeit('npo._copyto(npo.sim_matrix, npo.levels_array)', setup="import numpy_overwrite as npo;", number=3))
print('_just_assign')
print(timeit.timeit('npo._just_assign(npo.sim_matrix, npo.levels)', setup="import numpy_overwrite as npo;", number=3))
print(timeit.timeit('npo._just_assign(npo.sim_matrix, npo.levels_array)', setup="import numpy_overwrite as npo;", number=3))

---

运行测试

_fromiter
=========
from python list: 21.129429172957316
from numpy.array: 61.76275303697912

_copyto
=========
from python list: 52.4522930260282
from numpy.array: 0.7605530479922891

_just_assign
=========
from python list: 52.251478374004364
from numpy.array: 0.6351132979616523
    

Answer 1

您可能只是在寻找这个吗？

sim_matrix[:] = levels

EDIT：要修改一行，应该使用sim_matrix[row_id] = levels

另外，请注意using mutables as default function arguments。