等效于内联函数或宏的Python

Question

我只是意识到

x.real*x.real+x.imag*x.imag

比做的快三倍

abs(x)**2

其中x是一个复数的numpy数组。为了提高代码的可读性，我可以定义一个类似

def abs2(x):
    return x.real*x.real+x.imag*x.imag

仍然比abs（x）** 2快得多，但这是以函数调用为代价的。是否可以像在C中使用宏或使用inline关键字那样内联这样的函数？

Answer 1

是否可以像在C中使用宏或使用inline关键字那样内联这样的函数？

不会。在到达此特定说明之前，Python解释器甚至不知道是否存在这样的函数，更不用说它做了什么。

如评论中所述，PyPy 将自动内联
（上面仍然成立-它“简单地”在运行时生成优化的版本，从中受益，但在无效时中断它），尽管在这种情况下这无济于事因为在PyPy上实施NumPy只是在不久前才开始，直到今天还不是beta版。但最重要的是：不要担心在Python上进行此级别的优化。实现本身可以对其进行优化，也可以不进行优化，这不是您的责任。