用python写一个绘制单双叶曲面

在Python中，我们可以使用matplotlib库的`mplot3d`模块以及numpy库来绘制单叶和双叶曲面。这里是一个简单的例子：

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

# 定义函数生成数据
def single_leaf(u, v):
    return np.sin(np.pi * u) * np.cos(2 * np.pi * v)

def double_leaf(u, v):
    return np.sin(np.pi * (u**2 + v**2))

# 创建网格
u = np.linspace(-np.pi, np.pi, 100)
v = np.linspace(-np.pi, np.pi, 100)
U, V = np.meshgrid(u, v)

# 绘制单叶曲面
ax1 = plt.figure().add_subplot(projection='3d')
ax1.plot_surface(U, V, single_leaf(U, V), cmap='viridis', antialiased=True)
ax1.set_xlabel('X')
ax1.set_ylabel('Y')
ax1.set_zlabel('Z')
ax1.title.set_text('Single Leaf')

# 绘制双叶曲面
ax2 = ax1.twin()
ax2.plot_surface(U, V, double_leaf(U, V), cmap='inferno', alpha=0.7)
ax2.set_title('Double Leaf')

plt.show()