你了解matplotlib嗎?多類別條形圖篇

你了解matplotlib嗎?多類別條形圖篇

這篇文章來探索下多類別條形圖比如各學校包含語文、數學、英語三科成績的條形圖怎樣繪製。在繪圖之前,先來複習一下條形圖函數中主要參數的含義:

  • x控制的是每個條在x軸方向上的位置;
  • height控制的是每個條的長度;
  • width控制的是每個條的寬度;
  • bottom控制的是每個條在y軸方向的起始位置
  • orientation控制的是條形的方向,是縱向還是橫向,默認是縱向的。

小例子輔助理解:

x = [0.7, 1.5, 2, 3]height = [3, 10, 12, 7]plt.bar(x, height, width=0.3, bottom=[3, 0, 0, 1] )plt.show()
orientation控制的是條形的方向,是縱向還是橫向,默認是縱向

參照代碼和圖形再理解下各個參數的作用。ok,萬事俱備,開始繪圖!

先看一下原數據data1長什麼樣子再敲代碼:

寬度;bottom控制的是每個條在y軸方向的起始位置
plt.figure(figsize=(16,6))x_s = np.array(list(range(0,20,2))) #設置語文成績的條形所在位置plt.bar(x_s,data1.loc[:,"語文"].iloc[:10],width=0.5) #繪製語文成績的條形圖x_y = np.array(list(range(0,20,2)))+0.5 #設置數學成績的條形所在位置plt.bar(x_y,data1.loc[:,"數學"].iloc[:10],width=0.5) #繪製數學成績的條形圖x_y = np.array(list(range(0,20,2)))+1 #設置英語成績的條形所在位置plt.bar(x_y,data1.loc[:,"英語"].iloc[:10],width=0.5) #繪製英語成績的條形圖plt.title("成績條形圖",fontsize = 14)plt.ylabel("成績",fontsize = 14)plt.xticks(x_s+0.5,data1.iloc[:,0].iloc[:10],fontsize = 12) #x軸刻度為各學校名稱,為了刻度正好在三個條形的正中間,設置(x_s+0.5)plt.legend(["語文","數學","英語"]); 

​ 坐標軸和rc參數設置的講解中有提到過,在同一塊畫布上是可以重複繪圖的,其實在一幅條形圖中繪製多個類別的條形圖應用的就是這個原理。需要注意的是不要讓後邊繪製的圖形覆蓋前邊繪製的圖形,所以需要提前計算好每個條形應該畫在哪個地方。

​ 通過代碼可以看到都進行了哪些設置,每個條形的寬度都是0.5,所以在繪製好第一個學科的條形圖後,排在第二個位置進行繪製的條形圖所有的條所在x軸的位置都在第一個學科條形位置的基礎上增加了0.5,第三個學科的條形又在第二個學科條形位置的基礎上再增加0.5的距離,這樣,每個學校的三個學科可以挨著展示出來又不會發生條形重疊的情況。

​ 這裡需要注意的另一個問題就是每個學校刻度之間要流出足夠的位置放置三個條形。每個條形的寬度都是0.5,一共需要1.5個位置,所以在設置刻度的時候,每個刻度之間的間隔(range(0,20,2)),一共10個刻度,對應選取的10所學校,刻度間距為2,超過所需的1.5。

​ 最終的效果圖:

height控制的是每個條的長度;width控制的是每個條的

​ 圖是畫出來了,但是畫的過程中需要心算一下各種位置,除了各個條形的位置還有刻度的位置,著實有點麻煩,那麼有沒有其他更簡便一點的方法呢?

DataFrame繪圖

​ 先來和我們熟悉的matplotlib繪圖做對比,用matplotlib繪圖時先選擇繪圖函數,然後把原數據作為參數傳入函數中,而dataframe直接繪圖的時候,類似調用了dataframe的方法,在通過參數選擇進行哪種圖形的繪製。

​ 干說沒實感,來段代碼感受下:

 #通過參數設置生成圖形的類型data2.iloc[:10].plot(x = '學校',y = ['語文','數學','英語'],kind = "bar",figsize=(16,6),width=0.7,rot = 0,title = "各學科成績條形圖");

一行代碼搞定,先來看下效果圖:

:x控制的是每個條在x軸方向上的位置;

是不是看起來和上邊matplotlib繪製的圖差不多,代碼卻簡潔了很多。先看下原dataframe長什麼樣子:

成績的條形圖怎樣繪製。在繪圖之前,先來複習一下條形圖函數中主要參數的含義

接著具體研究下都是哪些參數在影響繪圖:

  • x和y沒有什麼懸念,指定x軸和y軸數據來源,其中x的數據源是學校名稱,y軸是各個學科成績
  • kind參數控制的是繪製什麼類型的圖,這裡選擇是「bar」,即表示繪製條形圖,當然還有其他的折線圖、餅圖、直方圖等等類型可以選擇
  • 其他的參數和matplotlib中的參數設置差不多,個別的參數會有差別,比如設置標籤字體傾斜度的,在matplotlib中參數是rotation,在dataframe中的參數是rot,寫法有點不一樣,作用是一樣的,所以雖然沒有在代碼中進行傾斜度的設置,還是把這個參數寫出來了,設置的傾斜度是0度,也就是沒傾斜。

這裡寫出兩種方法實現相同的操作,在實際的工作中,按照需求自行選擇即可。

堆積條形圖

​ 有時候,除了查看單個類別的情況,還需要同時查看總體的情況,這就是堆積條形圖擅長的領域了。繪製堆積條形圖和繪製普通條形圖用的都是plt.bar()函數,也是通過參數設置實現堆積條形圖的繪製。

​ 還是對參數設置不太熟的孩子可以回到文章開頭回憶下重要參數的作用,這裡實現用語文、數學、英語三科成績的堆積條形圖:

plt.figure(figsize=(16,6))plt.bar(range(21),data1.loc[:,"語文"],width=0.9,label = "語文")plt.bar(range(21),data1.loc[:,"數學"],bottom=np.array(data1.loc[:,"語文"]),width=0.9,label ="數學")plt.bar(range(21),data1.loc[:,"英語"],bottom=np.array(data1.loc[:,"語文"]+data1.loc[:,"數學"]) ,width=0.9,label ="英語")totle_score = np.array(data1.loc[:,"語文"]+data1.loc[:,"數學"]+data1.loc[:,"英語"]).astype("int")for i in range(21): plt.text(i-0.25,totle_score[i]+1,totle_score[i]) #為條形圖中的每個條添加標籤plt.title("各學校成績堆積圖",fontsize = 14)plt.ylabel("成績",fontsize = 14)plt.xticks(range(21),data1.iloc[:,0],rotation=30,fontsize = 12)#x軸刻度為各學校名稱plt.legend() #顯示圖例;

​ 原理和前邊畫三科成績條形圖一樣,都是在同一塊畫布上重複繪圖,注意繪圖的邏輯即可。這裡是先畫一個學科成績的縱向條形圖,通過bottom參數控制第二個學科成績繪圖時在y軸方向的起始值,也就是在第一科學科成績條形的頂端接著畫第二科學科成績的條,然後在第二科成績條形的頂端繼續畫第三個學科成績的條形,這樣形成的就是堆積條形圖。

​ 當然還有一些細節需要注意,比如三個學科條形的寬度得設置成一樣的尺寸,避免影響美觀。

​ 細心的孩子可能已經發現了一點新鮮的設置:多了一個函數plt.text(),這個函數的作用是在條形的頂端添加了該條形的標籤,即三科的總成績。因為要對每個條形增加一個標籤,所以運用了循環,將總分作為標籤添加到了圖形中。

​ 通過plt.text()函數可以在整幅圖的任意位置添加需要的文本進去。函數的第一個參數控制文本在x軸方向的位置,第二個參數控制文本在y軸方向的位置,第三個參數為添加的文本信息。

​ 囉里囉唆說了這麼多,來看下堆積條形圖的效果:

這篇文章來探索下多類別條形圖比如各學校包含語文、數學、英語三科

​ 和常見的堆積條形圖沒什麼不一樣對吧,其實條形圖依然又值得深挖的地方,還有很多有意思的設置可以嘗試呢。來個預告,計劃在下一篇文章離探索一下發散型條形圖。

聲明:文章觀點僅代表作者本人,PTTZH僅提供信息發布平台存儲空間服務。
喔!快樂的時光竟然這麼快就過⋯
繼續其他精彩內容吧!
more