对于Python来说，计算复数的辐角是一个常见的需求。
一、cmath库中的phase函数
Python内置了一个cmath库，其中包括了计算复数辐角的函数 – phase。这个函数可以返回一个复数的辐角，单位是弧度。
import cmath # 定义一个复数 z = complex(1, 1) # 计算复数的辐角 theta = cmath.phase(z) # 输出结果 print(theta) # 0.7853981633974483
通过上面的代码可以看到，使用cmath库中的phase函数可以很容易地计算出复数的辐角，返回的结果是一个浮点数。
二、数学库中的atan2函数
除了使用cmath库内置的phase函数，还可以使用数学库中的atan2函数来计算复数的辐角。该函数返回的结果也是一个浮点数，单位是弧度。
import math # 定义一个复数 z = complex(1, 1) # 计算复数的辐角 theta = math.atan2(z.imag, z.real) # 输出结果 print(theta) # 0.7853981633974483
通过上面的代码可以看到，使用数学库中的atan2函数同样可以很容易地计算出复数的辐角，返回的结果也是一个浮点数。
三、使用numpy库做向量化计算
当需要对多个复数进行辐角计算时，可以使用numpy库来实现向量化计算。这种方法在处理复杂数据时很高效。
import numpy as np # 定义多个复数 z = np.array([1+1j, 2+2j, 3+3j]) # 计算复数的辐角 theta = np.angle(z) # 输出结果 print(theta) # [0.78539816 0.78539816 0.78539816]
通过上面的代码可以看到，使用numpy库中的angle函数可以对多个复数进行辐角计算，返回的结果是一个数组。
四、使用matplotlib库绘制极坐标图
除了计算复数的辐角，还可以使用matplotlib库来绘制复数的极坐标图。这种方法可以更直观地展示复数的辐角。
import matplotlib.pyplot as plt
# 定义一个复数
z = complex(-1, 1)
# 获取复数的模长和辐角
r = abs(z)
theta = cmath.phase(z)
# 绘制极坐标图
fig, ax = plt.subplots(subplot_kw={'projection': 'polar'})
ax.scatter(theta, r)
plt.show()
通过上面的代码可以看到，使用matplotlib库中的polar函数可以绘制复数的极坐标图，更直观地展示复数的辐角。