Python创建瀑布图教程

瀑布图是一种特殊的图表类型，常用于展示数据的变化情况，尤其是在分析数据的增减趋势时非常有用。在Python中，我们可以使用一些强大的绘图库来创建瀑布图，本教程将介绍如何使用matplotlib库来实现。

确保你已经安装了matplotlib库。如果没有安装，可以使用以下命令进行安装：

pip install matplotlib

接下来，我们将通过一个简单的示例来演示如何创建瀑布图。假设我们有一组数据，代表了某个项目在不同阶段的收支情况。

import matplotlib.pyplot as plt
import numpy as np

# 数据
stages = ['初始预算', '收入', '支出1', '收入2', '支出2']
amounts = [1000, 500, -300, 400, -200]

# 计算累计值
cumulative = np.cumsum(amounts)

# 创建瀑布图
fig, ax = plt.subplots()
for i in range(len(amounts)):
    if amounts[i] >= 0:
        ax.bar(i, amounts[i], bottom=cumulative[i] - amounts[i], color='g')
    else:
        ax.bar(i, amounts[i], bottom=cumulative[i], color='r')

# 设置图表标签和标题
ax.set_xticks(range(len(stages)))
ax.set_xticklabels(stages)
ax.set_ylabel('金额')
ax.set_title('项目收支瀑布图')

# 显示图表
plt.show()

在上述代码中，我们首先定义了数据阶段和对应的金额。然后，使用np.cumsum函数计算了累计值。接着，通过循环遍历数据，根据金额的正负来绘制不同颜色的柱状图，代表收入和支出。最后，设置了图表的标签和标题，并显示了瀑布图。

如果你想要进一步美化瀑布图，可以对图表的颜色、字体、线条等进行调整。例如，可以添加数据标签，使图表更加清晰易懂。

使用Python的matplotlib库可以方便地创建瀑布图。通过简单的代码，我们就能直观地展示数据的变化情况，为数据分析和决策提供有力支持。掌握了这个教程，你就可以在自己的项目中灵活运用瀑布图来展示数据啦。