numpy.arctan2 () in Python