Équivalence en Python des fonctions ou macros en ligne

Question

Je viens de réaliser que faire

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

est trois fois plus rapide que de faire

abs(x)**2

où x est un tableau numpy de nombres complexes. Pour la lisibilité du code, je pourrais définir une fonction telle que

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

ce qui est toujours beaucoup plus rapide que abs(x)**2, mais au prix d'un appel de fonction. Est-il possible de mettre en ligne une telle fonction, comme je le ferais en C en utilisant une macro ou le mot-clé inline ?

Answer 1

Vous pouvez essayer d'utiliser lambda :

abs2 = lambda x : x.real*x.real+x.imag*x.imag

puis l'appeler par :

y = abs2(x)