matplotlib 3d 散点图

from matplotlib import pyplot as plt;
from mpl_toolkits.mplot3d import Axes3D  
from matplotlib import cm  
import numpy as np

x=np.arange(1,6,0.1)
y=np.cos(x)
z=np.sin(x*x)

fig = plt.figure()	
cm = plt.get_cmap("RdYlGn")
col = [cm(float(i)/(30)) for i in range(100)]
ax3D = fig.add_subplot(111, projection='3d')
ax3D.scatter(x, y, z, s=10, c=col, marker='o')
plt.show()