将乘法结果保存到现有数组

Question

考虑以下代码：

a = numpy.array([1,2,3,4])
b = numpy.array([5,6,7,8])
# here a new array (b*2) will be created and name 'a' will be assigned to it
a = b * 2 

那么，numpy 是否可以将b*2 的结果直接写入已经为a 分配的内存，而不分配新的数组？

Answer 1

是的，这是可能的 - 您需要使用 np.multiply 及其 out 参数：

np.multiply(b, 2, out=a)

数组a 现在被b * 2 的结果填充（并且没有分配新的内存来保存函数的输出）。

NumPy 的所有 ufunc 都有 out 参数，这在处理大型数组时特别方便；通过允许重用数组，它有助于将内存消耗降至最低。唯一需要注意的是，数组必须具有正确的大小/形状才能保存函数的输出。