python绘制三维图形

import math
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
x=[i for i in range(1,100,1)]
# print(x)
y=[0.01*(i*math.log2(2*i/100)+(100-i)*math.log2(2*(100-i)/100)) for i in x]
# y=[2*i for i in x]
# plt.plot(y)
# plt.show()

np.seterr(divide='ignore', invalid='ignore') # 忽略0分母和nan
a=np.array(x[1:-2])
b=a
# print(b)
X,Y=np.meshgrid(a,b)
Z=0.01*(X*np.log2(2*X/100)+Y*np.log2(2*Y/100)+(100-X-Y)*np.log2(2*(100-X-Y)/100))
# print(Z.size)
z2=Z.tolist()
realZ=[]
for i in range(96):
    for j in range(96):
        if z2[i][j]>0.0001 :#
            realZ.append(z2[i][j])
        else: #过小值设为0，用来去除nan
            realZ.append(0)
print(realZ)
realZ=np.array(realZ)
realZ=realZ.reshape(96,96)#将一维变为二维
# c=0.01*(i*math.log2(2*i/100)+j*math.log2(2*j/100)+(100-i-j)*math.log2(2*(100-i-j)/100))
fig = plt.figure()  #定义新的三维坐标轴
ax3 = plt.axes(projection='3d')
ax3.plot_surface(X,Y,realZ,cmap='rainbow')
#ax3.contour(X,Y,Z, zdim='z',offset=-2，cmap='rainbow)   #等高线图，要设置offset，为Z的最小值
plt.show()