python - 绘制一个螺旋，其中 r=theta^2 for 0<= theta <= 10*pi ...in python

Question

所以我的问题是如何制作极坐标图 r = f(theta)对于一个函数f通过计算r获取一系列 theta 值，然后转换 r使用方程将 theta 和 theta 转换为笛卡尔坐标 x = r cos(theta) , y = r sin(theta) .

但是我需要绘制螺旋 r = (theta)^2对于 0 <= theta <= 10*pi这就是我到目前为止所拥有的……这里没有螺旋式上升。

#! /usr/bin/env python

import matplotlib.pyplot as plt
from math import cos, sin, pi
from numpy import linspace

for theta in linspace(0,10*pi):
        r = ((theta)**2)
        x = r*cos(theta)
        y = r*sin(theta)


plt.plot(x,y)  
plt.savefig("spiral.png")
plt.show()

Answer 1

您需要创建一个值列表，而不仅仅是一个点。在您的情况下，您不断计算 x 和 y，但永远不会将它们保存在任何地方。因此，您所绘制的只是最后一次迭代之后的对 (x,y)。

x = []
y = []
for theta in linspace(0,10*pi):
    r = ((theta)**2)
    x.append(r*cos(theta))
    y.append(r*sin(theta))

plt.plot(x,y)  
plt.show()

输出