也许是个开始,但我会检查matplotlib2.0看看3D是否更好import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
def Lin3dSeg(a,b):
'''
if a, b are spherical coordinate vector tuples, then it
returns list of spherical coordinate points of a linear in r, theta, phi
3D spiral segment connecting them
'''
return [np.linspace(start, stop) for start, stop in zip(a,b)]
def Sphere2Cart(sphere_pt): # no particular convention, just thrown together
x = sphere_pt[0] * np.sin(sphere_pt[1])
y = sphere_pt[0] * np.cos(sphere_pt[1])
z = sphere_pt[0] * np.cos(sphere_pt[2])
return (x, y, z)
def SphereSegCarts(seg):
return [Sphere2Cart(pt) for pt in zip(*Lin3dSeg(*seg))]
seg_a =((10, 0, 0),(5, np.pi/2, 0))
seg_b =((10, np.pi/2, 0),(1, 0, np.pi/2))
seg_c =((3, 0, np.pi/2),(10, np.pi/2, np.pi/2))
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.view_init(elev=45, azim=45)
ax.plot(*zip(*SphereSegCarts(seg_a)), c='r')
ax.plot(*zip(*SphereSegCarts(seg_b)), c='g')
ax.plot(*zip(*SphereSegCarts(seg_c)), c='b')
plt.show()