三维瀑布图python

要绘制三维瀑布图，同样可以使用Python中的Matplotlib库的mplot3d子库。下面是一个简单的绘制三维瀑布图的示例代码：

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

# 生成数据
x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)
x, y = np.meshgrid(x, y)
r = np.sqrt(x ** 2 + y ** 2)
z = np.sin(r)

# 绘制图形
fig = plt.figure()
ax = Axes3D(fig)

# 绘制底部的平面
ax.plot_surface(x, y, np.zeros_like(z), rstride=1, cstride=1, alpha=0.2)

# 绘制瀑布图
levels = np.linspace(z.min(), z.max(), 30)
for i, level in enumerate(levels):
    ax.contourf(x, y, z, zdir='z', offset=level, cmap='rainbow')

# 显示图形
plt.show()

运行以上代码，就可以得到一个简单的三维瀑布图。其中，x、y、z是我们要绘制的数据，r是用来计算z值的辅助变量，这里使用了numpy库的meshgrid函数来生成x、y的坐标网格。绘图部分则是使用了Matplotlib库的mplot3d子库中的plot_surface和contourf函数来绘制三维瀑布图，其中zdir和offset参数用于控制瀑布图在z轴上的位置和高度，cmap参数用于设置颜色映射。