1. 程式人生 > >seaborn在柱狀圖上顯示y軸的數值

seaborn在柱狀圖上顯示y軸的數值

import seaborn as sns
import numpy as np
sns.set(style="ticks",color_codes=True)
import matplotlib.pyplot as plt
#正常顯示中文
plt.rcParams["font.sans-serif"]=["SimHei"]

if __name__ == "__main__":
    #獲取資料集
    data = sns.load_dataset("tips")
    #分組求和
    grouped_values = data.groupby("day").sum().reset_index()
    #設定繪製柱狀圖的顏色
    pal = sns.color_palette("Greens_d",len(grouped_values))
    rank = grouped_values["total_bill"].argsort().argsort()
    g = sns.barplot(x="day",y="tip",data=grouped_values,palette=np.array(pal[::-1])[rank])
    #在柱狀圖的上面顯示各個類別的數量
    for index,row in grouped_values.iterrows():
        #在柱狀圖上繪製該類別的數量
        g.text(row.name,row.tip,round(row.total_bill,2),color="black",ha="center")
    plt.show()