如何用python绘制瀑布图

在Python中，我们可以使用matplotlib库来绘制瀑布图。下面是一个简单的例子：

import matplotlib.pyplot as plt

# 数据
labels = ['A', 'B', 'C', 'D', 'E']
values = [10, 15, 5, -7, 20]

# 计算每个数据点的起始位置和终止位置
cumulative = 0
positions = []
for value in values:
    positions.append(cumulative)
    cumulative += value
positions.append(cumulative)

# 绘制瀑布图
plt.bar(labels, values, color='b')
plt.plot(positions, linestyle='--', color='r')
plt.xticks(rotation=45, ha='right')
plt.show()

这个例子中，我们定义了一个包含5个数据点的瀑布图，每个数据点用一个字母标识。我们使用`bar`函数绘制每个数据点的柱形图，使用`plot`函数绘制分割线。

为了计算每个数据点的起始位置和终止位置，我们定义了一个`cumulative`变量，用于记录前面数据点的累加和。在循环中，我们将每个数据点的起始位置设为前面所有数据点的累加和，终止位置设为当前数据点的值加上前面所有数据点的累加和。

最后，我们使用`xticks`函数将标签旋转45度并向右对齐，以便更好地显示标签。