import numpy as np import matplotlib.pyplot as plt # 极坐标螺旋线 r = np.arange(0,3,0.002) theta = 2 * np.pi * r fig = plt.figure(figsize=(8, 8)) ax = fig.add_subplot(111, projection='polar') ax.plot(theta, r) ax.set_rticks([0.5, 1, 1.5, 2, 2.5, 3, 4]) fig, ax = plt.subplots(subplot_kw={'projection': 'polar'}) ax.plot(theta, r, label='Archimedean spiral') ax.plot(theta_intersect, r_intersect, 'ro', label='Intersection point') ax.annotate('Intersection point', xy=(theta_intersect, r_intersect), xytext=(theta_intersect+0.5, r_intersect+0.5), arrowprops=dict(facecolor='black', shrink=0.01), fontsize=10, horizontalalignment='right', verticalalignment='bottom') # 添加注释 ax.annotate('Start Here', xy=(0, 0), xytext=(0.1, 1.2), arrowprops=dict(facecolor='black', shrink=0.05)) ax.annotate('End Here', xy=(10*np.pi, 10*np.pi), xytext=(9.5, 9.5), arrowprops=dict(facecolor='black', shrink=0.05)) ax.legend() plt.show()代码优化

可以考虑将两个子图合并成一个，同时移除不必要的代码注释。代码优化后如下：

import numpy as np 
import matplotlib.pyplot as plt  

r = np.arange(0, 3, 0.002)  
theta = 2 * np.pi * r 

fig = plt.figure(figsize=(8, 8)) 
ax = fig.add_subplot(111, projection='polar') 

# 绘制极坐标螺旋线
ax.plot(theta, r, label='Archimedean spiral') 

# 设置极坐标刻度
ax.set_rticks([0.5, 1, 1.5, 2, 2.5, 3, 4]) 

# 添加交点的标记
theta_intersect = np.pi / 4 
r_intersect = theta_intersect / (2 * np.pi)
ax.plot(theta_intersect, r_intersect, 'ro', label='Intersection point') 
ax.annotate('Intersection point', xy=(theta_intersect, r_intersect), xytext=(theta_intersect+0.5, r_intersect+0.5),             
            arrowprops=dict(facecolor='black', shrink=0.01), fontsize=10, horizontalalignment='right', verticalalignment='bottom') 

# 添加起点和终点的标记
ax.annotate('Start Here', xy=(0, 0), xytext=(0.1, 1.2), arrowprops=dict(facecolor='black', shrink=0.05)) 
ax.annotate('End Here', xy=(10*np.pi, 10*np.pi), xytext=(9.5, 9.5), arrowprops=dict(facecolor='black', shrink=0.05)) 

# 添加图例
ax.legend() 

plt.show()