Matplotlib：将 plot_surface 中的 cmap 设置为 x 和 y 轴

Question

如何设置与图形半径相关的颜色图？我怎样才能关闭圆柱体的末端（在元件上，而不是顶部和底部）？我的脚本：import numpy as npfrom matplotlib import pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dfrom math import sin, cos, pifig = plt.figure()ax = fig.add_subplot(111, projection='3d')h, w = 60,30znew = np.random.randint(low=90, high=110, size=(60,30))theta = np.linspace(0,2*pi, h)Z = np.linspace(0,1,w)Z,theta = np.meshgrid(Z, theta)R = 1X = (R*np.cos(theta))*znewY = (R*np.sin(theta))*znewax1 = ax.plot_surface(X,Y,Z,linewidth = 0, cmap="coolwarm",              vmin= 80,vmax=130, shade = True, alpha = 0.75)fig.colorbar(ax1, shrink=0.9, aspect=5)plt.show()