首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >Python函数:返回一个没有额外内存分配的数组

Python函数:返回一个没有额外内存分配的数组
EN

Stack Overflow用户
提问于 2018-07-26 04:28:13
回答 2查看 667关注 0票数 3

假设我想做一个用输入向量乘以输入矩阵的函数:

代码语言:javascript
复制
def MatMul(A,b):
    return A.dot(b)

现在,我执行以下代码:

代码语言:javascript
复制
import numpy as np
A=np.array([[1,2,3],[4,5,6],[7,8,9]],dtype='float64')
b=np.array([4,5,6],dtype='float64')
c=np.zeros(3,dtype='float64')
c=MatMul(A,b)

MatMul函数中会有额外的数组分配吗?我知道Ab将通过引用传递。注意,我已经预先分配了数组c

通常,如何在这样的简单函数中避免不必要的预分配?比如说,我想执行几个数学运算:

代码语言:javascript
复制
def Rank1Update(A,b,alpha):
    c=A.dot(b)
    c+=alpha*c.dot(c)*c
    return c

我可以在一行中包含很多数学函数,但是代码很快就变得不可读了。

我熟悉C风格的编程,在这种编程中,为了避免不必要的内存分配,可以通过引用传递Abc,并在返回void的函数中更新c。我可以在python中做同样的事情,但为了方便和代码可读性,我想使用return

谢谢,

米哈伊尔

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2018-07-26 04:35:20

Numpy在它的大多数生成新数组的函数中,都有一个参数来存储结果数组。我在下面的代码中使用了该参数的名称版本out,但您可以省略该名称。您必须确保out数组具有正确的形状和数据类型。这个参数的目的正是您想要的--避免额外的内存分配。这也可以加快代码的速度。

代码语言:javascript
复制
import numpy as np
A = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype='float64')
b = np.array([4, 5, 6], dtype='float64')
c = np.zeros(3, dtype='float64')

A.dot(b, out=c)

documentation for dot()中提到了该参数。如果您想要更改函数MatMul的定义,

代码语言:javascript
复制
def MatMul(A, b, c=None):
    return A.dot(b, out=c)

并将调用更改为

代码语言:javascript
复制
MatMul(A, b, c)
票数 3
EN

Stack Overflow用户

发布于 2018-07-26 04:34:33

dot不知道也不关心c变量或该变量已经包含引用的数组。它将生成一个新的数组,=会将c变量绑定到这个新的数组,让内存管理系统清理旧的数组。

如果您希望dot将输出存储到现有数组中,则需要告诉它这样做:

代码语言:javascript
复制
A.dot(b, out=c)
票数 3
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/51527164

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档