如果我們想要展示某些數據,柱狀圖是一個常用的作圖方式。在Python的matplotlib庫中,plt.bar()是一個非常方便的函數,可以快速繪製柱狀圖。在本文中,我們將對plt.bar()進行各個方面的詳細講解。
一、plt.barh()
plt.barh()是plt.bar()的水平版,它可以繪製橫向的柱狀圖。它的用法與plt.bar()類似,只是將x變數替換為y變數,將width變數替換為height變數:
import matplotlib.pyplot as plt
import numpy as np
x = ["A", "B", "C", "D", "E"]
y = [20, 35, 30, 25, 15]
plt.barh(y=y, width=x, height=0.5)
plt.show()
在這個例子中,我們使用了一個包含5個元素的列表x作為寬度(width),另一個包含5個元素的列表y作為高度(height)。我們將列表x賦值給y,將列表y賦值給height,這樣就可以實現橫向的柱狀圖。代碼執行後,將會展示以下圖像:
二、plt.bar()的用法
1.繪製基本柱狀圖
使用plt.bar()繪製基本的柱狀圖非常簡單,只需要傳入x和height兩個參數即可。例如,我們想要繪製一個長為5的柱狀圖,其高度分別為20、35、30、25和15,可以這樣實現:
import matplotlib.pyplot as plt
import numpy as np
x = [1, 2, 3, 4, 5]
heights = [20, 35, 30, 25, 15]
plt.bar(x=x, height=heights)
plt.show()
執行這段代碼後,將會繪製如下圖像:
2.設置柱狀圖顏色和邊框
對於柱狀圖,我們通常需要對其顏色進行調整。使用plt.bar()函數也可以非常方便地實現這一功能。我們可以通過color參數來設置柱狀圖的顏色,通過edgecolor參數來設置柱狀圖的邊框顏色。例如,我們可以繪製一組紅色的柱狀圖,並將邊框顏色設置為黑色:
import matplotlib.pyplot as plt
import numpy as np
x = ["A", "B", "C", "D", "E"]
heights = [20, 35, 30, 25, 15]
plt.bar(x=x, height=heights, color="red", edgecolor="black")
plt.show()
執行這段代碼後,將會繪製如下圖像:
3.設置柱狀圖寬度
有時候,我們可能需要調整柱狀圖的寬度。使用plt.bar()函數,我們可以通過設置width參數來實現這個功能。例如,如果我們想要繪製一組寬度為0.4的水平柱狀圖,可以這樣實現:
import matplotlib.pyplot as plt
import numpy as np
x = ["A", "B", "C", "D", "E"]
heights = [20, 35, 30, 25, 15]
plt.bar(x=x, height=heights, width=0.4)
plt.show()
執行這段代碼後,將會繪製如下圖像:
4.設置柱狀圖的標籤
在柱狀圖中,我們通常希望為每一個柱子添加一個標籤,例如柱子的名稱或其高度值。使用plt.bar()函數,我們可以通過在循環中對每一個柱子添加一個文本標籤來實現這個功能。例如,如果我們想要為每一根柱子添加一個高度標籤,可以這樣實現:
import matplotlib.pyplot as plt
import numpy as np
x = ["A", "B", "C", "D", "E"]
heights = [20, 35, 30, 25, 15]
plt.bar(x=x, height=heights, width=0.4)
for i in range(len(x)):
plt.text(x=i, y=heights[i]+1, s=heights[i])
plt.show()
執行這段代碼後,將會繪製如下圖像:
三、總結
在本文中,我們對plt.bar()函數進行了詳細的闡述,介紹了plt.barh()、顏色設置、寬度設置和標籤設置等內容。通過掌握這些知識,我們可以輕鬆地繪製出漂亮的柱狀圖。
原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-tw/n/228769.html