matplotlib绘制三维瀑布图

好的，下面是使用matplotlib绘制三维瀑布图的代码示例：

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

# 生成随机数据
x_data = np.arange(1, 6)
y_data = np.arange(1, 6)
z_data = np.array([[1, 2, 3, 4, 5],
                   [2, 3, 4, 5, 6],
                   [3, 4, 5, 6, 7],
                   [4, 5, 6, 7, 8],
                   [5, 6, 7, 8, 9]])

# 绘制三维瀑布图
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

for i in range(len(x_data)):
    ax.bar(x_data[i], y_data, zs=z_data[:, i], zdir='y', alpha=0.8, color='tab:blue')

# 设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

plt.show()

运行以上代码，即可得到一个简单的三维瀑布图。其中，`x_data`和`y_data`表示瀑布图上每个柱子的位置和高度，`z_data`表示每个柱子上方的高度。通过`ax.bar`函数绘制柱状图，其中`zs`参数表示每个柱子上方的高度，`zdir`参数表示柱子的方向，这里选择`y`方向。然后，通过设置坐标轴标签，即可完成三维瀑布图的绘制。