python 用 matplotlib 制作瀑布圖步驟詳解

1、加載庫

import numpy as np import pandas as pd import matplotlib as mpl import matplotlib.pyplot as plt

2、繪圖

# 配置字體，顯示中文
mpl.rcParams['font.sans-serif'] = ['SimHei'] # 配置坐標軸刻度值模式，顯示負號
mpl.rcParams['axes.unicode_minus'] = True # 定義數據
x = np.array([1, 2, 3, 4, 5, 6, 7])*2 y1 = np.array([0.5, 0.0, 1.4, 2.7, 1.5, 3.3, 1.5]) y2 = np.array([0.4, 0.5, 0.3, 0.3, 0.4, 0.3, 0.25]) y3 = np.array([20.66, 21.5, 23, 7, 24.5, 32, 20]) y4 = np.array([0.7, 0.4, 2.2, 0.9, 0.7, 0.3, 0.5]) y5 = np.array([3, 16, 12.2, 13.6, 10, 50, 19]) labels = list('ABCDEFG') w = 0.2 fig, ax = plt.subplots(figsize=(12, 8), facecolor='cornsilk') # 繪制柱狀圖 ## 第 1 層柱狀圖 
i = 0 ax.bar(x=x,    # 柱體在 x 軸上的坐標位置
       bottom=0, height=y1,    # 柱體的高度
       width=w, color='dodgerblue', # edgecolor='m',
 ) ## 第 2 層柱狀圖
i = 1 ax.bar(x= x+w*i,  # 柱體在 x 軸上的坐標位置
       bottom=y1, height=y2,    # 柱體的高度
       width=w, color='orange', # edgecolor='g',
 ) ## 第 3層柱狀圖
i = 2 ax.bar(x= x+w*i,    # 柱體在 x 軸上的坐標位置
       bottom=y1+y2, height=y3,    # 柱體的高度
       width=w, color='gray', alpha=0.5, # edgecolor='g',
       tick_label=labels, ) ## 第 4 層柱狀圖
i = 3 ax.bar(x= x+w*i,    # 柱體在 x 軸上的坐標位置
       bottom=y1+y2+y3, height=y4,    # 柱體的高度
       width=w, color='orange', # alpha=0.5, # edgecolor='g',
 ) ## 第 5 層柱狀圖
i = 4 ax.bar(x= x+w*i,    # 柱體在 x 軸上的坐標位置
       bottom=y1+y2+y3+y4, height=y5,    # 柱體的高度
       width=w, color='dodgerblue', # alpha=0.5, # edgecolor='g',
 ) for i, j in zip(x+w*3, y1+y2+y3+y4+y5+1): ax.text(x=i, y=j, s=f'{j:.2f}', size=12, color='black', family='Arial') ax.set_yticks(ticks=range(0, 101, 10)) # 添加圖例 # ax.legend()

# 設置網格線
ax.grid(axis='y', ls=':'
# alpha=0.6
       )

圖形

 

3、用 DataFrame 和 for 循環簡化上面的代碼

# 配置字體，顯示中文
mpl.rcParams['font.sans-serif'] = ['SimHei'] # 配置坐標軸刻度值模式，顯示負號
mpl.rcParams['axes.unicode_minus'] = True # 定義數據
x = np.arange(1, 15, 2) y1 = [0.5, 0.0, 1.4, 2.7, 1.5, 3.3, 1.5] y2 = [0.4, 0.5, 0.3, 0.3, 0.4, 0.3, 0.25] y3 = [20.66, 21.5, 23, 7, 24.5, 32, 20] y4 = [0.7, 0.4, 2.2, 0.9, 0.7, 0.3, 0.5] y5 = [3, 16, 12.2, 13.6, 10, 50, 19] df = pd.DataFrame(np.array([y1, y2, y3, y4, y5]).T) cs = ['dodgerblue', 'orange', 'lightgray', 'orange', 'dodgerblue'] w = 0.2 fig, ax = plt.subplots(figsize=(12, 8), facecolor='cornsilk') # 繪制柱狀圖
for i, c in enumerate(cs) : ax.bar(x=x+w*i,    # 柱體在 x 軸上的坐標位置
           bottom=df.iloc[:, :i].sum(axis=1), height=df[i],    # 柱體的高度
           width=w, color=c, ) # 添加數據標簽
for i, j in zip(x+w*3, df.sum(axis=1)+1): ax.text(x=i, y=j, s=f'{j:.2f}', size=12, color='black', family='Arial') ax.set_yticks(ticks=range(0, 101, 10)) # 設置網格線
ax.grid(axis='y', ls=':' )

圖形：