在python中用三轴网格绘制四维数据

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

# 创建四维数据
x = np.linspace(-5, 5, 100)  # x轴数据范围
y = np.linspace(-5, 5, 100)  # y轴数据范围
X, Y = np.meshgrid(x, y)  # 创建网格
Z = np.sin(np.sqrt(X**2 + Y**2))  # 计算z轴数据
W = np.cos(np.sqrt(X**2 + Y**2))  # 计算第四维数据

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X, Y, Z, cmap='viridis')  # 绘制三维曲面
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.show()