[surface plot] draw a surface using matplotlib #python

F1 Check.py

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

fig = plt.figure()
ax = Axes3D(fig)
X = np.arange(0, 1, 0.05)
Y = np.arange(0, 1, 0.05)
X, Y = np.meshgrid(X, Y)
#F1
Z = (2 * X * Y) / (X + Y)
Z[0][0] = 0

# #AVG
# Z = (X + Y) / 2

# #GEO AVG
# Z = np.sqrt(X * Y)

# 具体函数方法可用 help(function) 查看，如：help(ax.plot_surface)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')

plt.show()