首页
学习
活动
专区
工具
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
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/51527164

复制
相关文章

相似问题

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