Vectorized Beispiel der Math.atan2() Funktion in Python

Question

lassen Sie uns sagen, ich habe eine numpy Matrix A die Nx2 der Größe ist. Was ich tue, ist die Berechnung der 4-Quadranten-Arkustangens der ersten Säule und der zweiten Säule, wie so:

import math 
for i in xrange(A.shape[0]): 
    phase[i] = math.atan2(A[i,0], A[i,1]) 

ich aber dies in einer vektorisierten Weise tun möchten. Wie kann ich das machen? Die Funktion math.atan2() scheint die Vektorisierung nicht zu unterstützen.

Danke!

Answer 1

Es sieht für mich wie es nur sein sollte:

import numpy as np 
phase = np.arctan2(A[:, 0], A[:, 1]) 

Oder vielleicht (wenn phase eine andere Länge als A aus irgendeinem Grunde ist):

phase[:len(A)] = np.arctan2(A[:, 0], A[:, 1]) 

Mit anderen Worten, don ‚t verwenden math.atan2, verwenden numpy.arctan2 seit numpy Funktionen im allgemeinen Versionen ihrer math Pendants vektorisiert werden.