餅圖是數據可視化的常用方式之一,它能夠直觀地呈現出數據之間的比例關係。在Python中,使用matplotlib庫中的pyplot模塊可以輕鬆地畫出餅圖。
一、基本餅圖
使用matplotlib.pyplot.pie()函數可以簡單地畫出基本的餅圖。該函數有兩個必傳參數:sizes和labels,分別代表比例和標籤。請看下面的代碼:
import matplotlib.pyplot as plt sizes = [30, 40, 20, 10] labels = ['A', 'B', 'C', 'D'] plt.pie(sizes, labels=labels) plt.show()
上述代碼會生成一個基本的餅圖,其中比例從sizes中取得,標籤從labels中取得。plt.show()函數用於顯示圖像,如果省略該函數,則圖像無法顯示。
二、設置顏色和字體
默認情況下,matplotlib會為每一塊自動生成顏色。如果需要自定義顏色,可以使用colors參數。colors參數應該是一個長度等於sizes長度的元組,其中每個元素代表一種顏色。請看下面的代碼:
import matplotlib.pyplot as plt sizes = [30, 40, 20, 10] labels = ['A', 'B', 'C', 'D'] colors = ('red', 'blue', 'green', 'yellow') plt.pie(sizes, labels=labels, colors=colors) plt.show()
上述代碼中,自定義了四個顏色,分別對應四個標籤。也可以使用列表生成式生成自己的顏色列表。比如下面生成一個顏色漸變的列表:
import matplotlib.pyplot as plt from matplotlib import cm # 導入漸變色模塊 sizes = [30, 40, 20, 10] labels = ['A', 'B', 'C', 'D'] colors = [cm.Spectral(x) for x in np.linspace(0, 1, len(sizes))] # 生成顏色漸變列表 plt.pie(sizes, labels=labels, colors=colors) plt.show()
該方法使用了matplotlib中的colormap模塊,通過np.linspace()函數生成一組等差數列,再傳入原始顏色中,就可以生成一個顏色漸變的餅圖。
此外,可以使用字體庫改變文本的字體和大小。請看下面的代碼:
import matplotlib.pyplot as plt from matplotlib import font_manager as fm # 導入字體模塊 sizes = [30, 40, 20, 10] labels = ['A', 'B', 'C', 'D'] prop = fm.FontProperties(fname="font.ttf", size=14) # 定義字體樣式對象 plt.pie(sizes, labels=labels, textprops={'fontproperties':prop}) plt.show()
上述代碼中,創建了一個字體樣式對象prop,並將其設置為textprops參數的值,從而改變了餅圖中文本的字體和大小。其中,fname參數指定了字體文件路徑,size參數指定了字體大小。
三、強調特定部分
強調餅圖中的某一部分可以使用explode參數。該參數應該是一個長度等於sizes長度的元組,其中每個元素代表相應餅塊的偏移量。請看下面的代碼:
import matplotlib.pyplot as plt sizes = [30, 40, 20, 10] labels = ['A', 'B', 'C', 'D'] explode = (0, 0.1, 0, 0) # 強調B plt.pie(sizes, labels=labels, explode=explode) plt.show()
上述代碼中,通過設置explode參數,將餅圖中B部分的偏移量設置為0.1,即向外凸出。
四、整體調整
在餅圖中,有許多其他的調整參數,比如刻度線、比例字體等。可以通過plt.axis()函數對圖像進行整體調整。
import matplotlib.pyplot as plt sizes = [30, 40, 20, 10] labels = ['A', 'B', 'C', 'D'] explode = (0, 0.1, 0, 0) plt.pie(sizes, labels=labels, explode=explode) # 隱藏刻度線 plt.axis('equal') # 修改比例字體 plt.rcParams['font.family'] = 'SimHei' plt.rcParams['font.size'] = '12' plt.show()
上述代碼中,使用plt.axis(‘equal’)函數隱藏了刻度線;使用plt.rcParams函數改變了字體樣式和大小。
五、總結
本文介紹了Python中如何使用matplotlib.pyplot模塊畫餅圖,包括基本餅圖、自定義顏色和字體、強調特定部分、整體調整等多方面。可以根據實際需要進行調整,使生成的圖像更加符合要求。
原創文章,作者:HBEJZ,如若轉載,請註明出處:https://www.506064.com/zh-hant/n/373320.html