使用Python绘制瀑布图
- 瀑布图
- 效果
- 代码
瀑布图
瀑布图(Waterfall Chart)是一种数据可视化工具,用于展示累积数值的变化,尤其适合于展示随时间或过程中的增减变化。它通常用于财务分析,如展示收入、支出和净利润的变化过程。瀑布图通过柱形图表示每个变化的步骤,并使用不同颜色区分正负变化,使得整个变化过程一目了然。
效果
代码
import matplotlib.pyplot as plt
import numpy as np# 示例数据
categories = ['Start', 'Product A', 'Product B', 'Product C', 'End']
values = [1000, 200, -300, 150, 0] # 最后一个值为0,表示结束点# 计算每一步的累计值
cumulative = np.cumsum(values)
cumulative = np.insert(cumulative, 0, 0) # 插入起始点
cumulative = cumulative[:-1] # 去掉最后一个点的累计值# 定义颜色
colors = ['green' if val >= 0 else 'red' for val in values]# 创建瀑布图
fig, ax = plt.subplots(figsize=(10, 6))# 绘制矩形
for i in range(len(values)):ax.bar(categories[i], values[i], bottom=cumulative[i], color=colors[i])# 添加标签
for i in range(len(values)):if values[i] >= 0:ax.text(i, cumulative[i] + values[i]/2, f"+{values[i]}", ha='center', va='center', color='black')else:ax.text(i, cumulative[i] + values[i]/2, f"{values[i]}", ha='center', va='center', color='black')# 设置标题和标签
ax.set_title('Waterfall Chart')
ax.set_xlabel('Categories')
ax.set_ylabel('Values')# 显示图表
plt.show()