python matplotlib 最简易懂实现三维图表可视化,(颜色渐变)

效果图:

代码:

 1 from matplotlib import pyplot as plt
 2 import numpy as np
 3 from mpl_toolkits.mplot3d import Axes3D
 4     fig = plt.figure()
 5     ax = Axes3D(fig)
 6     X = np.arange(-4, 4, 0.25)
 7     Y = np.arange(-4, 4, 0.25)
 8     X, Y = np.meshgrid(X, Y)
 9     R = np.sqrt(X ** 2 + Y ** 2)
10     Z = np.sin(R)
11     ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')
12     plt.show()

猜你喜欢

转载自www.cnblogs.com/smartisn/p/12914033.html