极坐标是描述平面点的一种方法,它通过极径和极角来描述一个点的位置。Python提供了许多画图库,如matplotlib,可以很方便地绘制极坐标图。
import numpy as np import matplotlib.pyplot as plt # 构建数据 theta = np.arange(0, 2*np.pi, 0.02) rho = 2*np.sin(4*theta) # 绘制图像 fig = plt.figure(figsize=(6,6)) ax = fig.add_subplot(111, projection='polar') ax.plot(theta, rho, color='r', linewidth=2) ax.set_title("Polar Coordinates") plt.show()
上述代码首先使用numpy库构建了theta和rho两个数组,分别表示极角和极径。接着使用matplotlib库中的subplot函数建立了一个极坐标系的子图,通过plot函数来将theta和rho数组绘制在子图上。最后设置图像的标题,并使用show函数显示图像。
我们也可以使用大量的可选参数来美化作图效果,比如:
ax.plot(theta, rho, color='r', linewidth=2, linestyle='--', label='sin(4θ)') ax.legend(loc='best')
上面的代码使用了linestyle参数设置了线段的样式,用label参数为图像添加图例,并使用legend函数显示图例。除此之外,还有许多其他可选参数,可以参见matplotlib的官方文档。