pyspark.sql.functions.
atan2
col1 – coordinate on y-axis
col2 – coordinate on x-axis
the theta component of the point (r, theta) in polar coordinates that corresponds to the point (x, y) in Cartesian coordinates, as if computed by java.lang.Math.atan2()
New in version 1.4.