如何在 Python 中的一个 3d 图上绘制循环中的多条线？

Question

基本上，我在 Python 中循环生成光线，并试图将它们全部绘制在同一张图上。它们都应该在半径为 0.1 的圆上。每条射线都应位于圆上的某个位置，该位置随 arg 的变化而变化，在本例中为 theta。另外，只是提一下（虽然我不认为它是相关的）我在这里做 OOP。

我得到了正确的光线，但我无法在同一个 3d 图上得到它们，而且我不确定我应该怎么做。我以为使用 plt.show() 会给我一个包含所有 24 条射线的图表，但它只绘制了 24 个图表。 这是相关的代码供参考：

r = 0.1
arg = 0
for i in range (0,24):
    arg += np.pi/12
    x = r*np.sin(arg)
    y = r*np.cos(arg)
    l = ray.Ray(r=np.array([x,y,0]),v=np.array([0.5,0,5]))
    c = ray.SphericalRefraction(z0 = 100, curv = 0.0009, n1 = 1.0, n2 = 1.5, ar = 5)
    c.propagate_ray(l)
    o = ray.OutputPlane(250)
    o.outputintercept(l)
    points = np.array(l.vertices())
    fig = plt.figure()
    ax = plt.axes(projection='3d')
    #ax = fig.add_subplot(1,2,1,projection='3d')
    #plt.plot(points[:,2],points[:,0])
    ax.plot3D(points[:,0],points[:,1],points[:,2])
plt.show()

Answer 1

扩展 Mercury 的评论，必须在循环外创建图形和 axes 对象。

import matplotlib.pyplot as plt
import numpy as np

r = 0.1
arg = 0

fig = plt.figure()
ax = plt.axes(projection='3d')
for i in range(0,24):
    arg += np.pi/12 * i
    v1 = r*np.sin(arg)
    v2 = r*np.cos(arg)
    # ...
    # using sample data
    x = []
    y = []
    z = []
    for j in range(2):
        x.append(j*v1)
        y.append(j*v2)
        z.append(j)
    # add vertex to the axes object
    ax.plot3D(x, y, z)
plt.show()