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参数，在使用大型数组时尤其方便;通过允许重用数组，有助于将内存消耗降至最低。唯一需要注意的是，数组必须具有正确的大小/形状才能保存函数的输出。