matplotlib绘制3D函数图像

import numpy as np
import matplotlib.pyplot as plt
import torchvision.utils

# 绘制3D函数图像

ax = plt.subplot(111, projection='3d')
x = np.arange(-3, 3, 0.25)
y = np.arange(-3, 3, 0.25)
x, y = np.meshgrid(x, y)
z = np.sin(y)

ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")
ax.set(zlim = (-1, 1))
ax.plot_surface(x, y, z)

plt.show()