matplotlib绘制瀑布图

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

import matplotlib.pyplot as plt
import numpy as np

# 生成随机数据
x_data = range(1, 6)
y_data = np.random.randint(low=1, high=11, size=5)
y2_data = np.random.randint(low=1, high=11, size=5)

# 绘制瀑布图
fig, ax = plt.subplots()
ax.bar(x_data, y_data, color='tab:blue')
ax.bar(x_data, -y2_data, color='tab:orange')

# 增加连接线
for i in range(len(x_data)):
    ax.plot([x_data[i], x_data[i]], [y_data[i], -y2_data[i]], color='gray', linestyle='--')

plt.show()

运行以上代码，即可得到一个简单的瀑布图。其中，`x_data`和`y_data`表示瀑布图上每个柱子的位置和高度，`y2_data`表示每个柱子下方的高度。通过`ax.bar`函数绘制柱状图，其中颜色参数可以通过`color`设置。然后，使用`ax.plot`函数绘制连接线，即可完成瀑布图的绘制。