python阿基米德螺线绘制

Python可以利用matplotlib库中的函数来绘制阿基米德螺线(Archimedean spiral)。阿基米德螺线是一种数学曲线，其特点是离原点越远的点，到中心的距离增长速度比旋转的角度快，形成螺旋状。下面是一个简单的例子：

import numpy as np
import matplotlib.pyplot as plt

# 定义阿基米德螺线函数
def archimedes_spiral(r, theta):
    x = r * np.cos(theta)
    y = r * np.sin(theta)
    return x, y

# 设置参数
r_min, r_max = 0, 50  # 螺线范围
theta_start, theta_end = 0, 2*np.pi  # 角度范围
num_points = 1000  # 绘制点的数量

# 创建θ数组
theta = np.linspace(theta_start, theta_end, num_points)

# 计算并绘制螺线
x, y = archimedes_spiral(np.arange(r_min, r_max), theta)
plt.plot(x, y)

# 设置图例和标题
plt.xlabel('X')
plt.ylabel('Y')
plt.title('阿基米德螺线')
plt.show()

运行这个代码，你会看到一条从起点开始向外扩展的螺旋线。你可以通过调整`r_min`, `r_max`, `theta_start`, 和 `theta_end` 来改变螺线的形状。