如何在 Python 中更改 Matplotlib 顏色條的大小?


視覺化工具是 Matplotlib 庫的重要組成部分。其中一個工具是顏色條。它顯示了繪圖中資料值和顏色之間的對映關係。為了調整顏色條的大小以使其更清晰可見或更好地適應繪圖,Matplotlib 中的 colorbar() 函式提供了幾個引數。

在本文中,我們將討論如何使用不同的方法更改 matplotlib 顏色條的大小。

什麼是 Matplotlib 以及如何安裝它?

Matplotlib 是一個廣泛用於在 Python 中建立靜態、動畫和互動式視覺化的庫。它有一個名為“pyplot”的模組,透過提供控制線型、字型屬性、格式化軸等功能,使繪圖變得容易。

我們可以使用以下命令安裝 matplotlib –

pip install matplotlib

什麼是 pyplot?

Pyplot 是 Matplotlib 庫的一個子庫。它通常用於資料分析和視覺化任務,因為它可以幫助使用者使用不同的引數輕鬆更改其繪圖的外觀。它是一種用更少的程式碼建立複雜圖形的簡單方法。它提供了各種選項,例如不同的顏色、標題、線型,允許您向繪圖新增文字、影像等。

Matplotlib 中的顏色條?

顏色條是標量值對映的視覺化表示。顏色條是一個條形,其中包含各種顏色,放置在 matplotlib 圖表的一側。它是圖表中顯示的顏色圖例。預設情況下,matplotlib 顏色條的位置在 matplotlib 圖表的右側。

顏色條中的顏色對映包含五種顏色。“rainbow”顏色對映在顏色條中包含七種顏色。

matplotlib 的 pyplot 模組中的 colorbar() 函式用於為繪圖建立顏色條,指示顏色比例。我們可以更改顏色條的方向,它可以是水平的或垂直的。

示例

在這個繪圖中,我們使用 Matplotlib 在散點圖中建立了垂直顏色條。命名了兩個列表“items”和“sales”,然後使用 plt.scatter() 函式為散點圖建立了一個繪圖。

X 軸標題為“ITEMS”,Y 軸標題為“SALES”,並將整個圖形標題為“SHOP MANGEMENT”。為了在繪圖中建立顏色條,我們使用了 plt.colorbar() 函式。

import matplotlib.pyplot as plt
items=["chocolate","toffees","cake","colddrinks","lays","biscuit","frooti"]
sales=[89000,23000,98777,12000,65000,11100,96000]
c=['r','y','m','r','y','m','b']
a=plt.scatter(items,sales,color=c)
plt.xlabel("ITEMS")
plt.ylabel("SALES")
plt.suptitle("SHOP MANGEMENT",fontsize=20)
plt.colorbar()
plt.show()

輸出

示例

在這個繪圖中,我們透過在 Matplotlib 的 plt.colorbar 函式中使用 orientation=”horizontal” 建立了水平顏色條。

import matplotlib.pyplot as plt
items=["chocolate","toffees","cake","colddrinks","lays","biscuit","frooti"]
sales =[89000,23000,98777,12000,65000,11100,96000]
c= ['r','y','m','r','y','m','b']
a=plt.scatter(items, sales, color=c)
plt.xlabel("ITEMS")
plt.ylabel("SALES")
plt.suptitle("SHOP MANGEMENT", fontsize=20)
plt.colorbar(orientation="horizontal")
plt.show()

輸出

熱力圖的顏色條

以類似的方式,您可以將顏色條新增到不同的繪圖中。讓我們以熱力圖繪圖為例。我們可以使用 NumPy rand() 函式建立一個 2D 隨機值陣列。

示例

import matplotlib.pyplot as plt
import numpy as np
data = np.random.rand(4, 4)
heatmap = plt.imshow(data, cmap='Greens')
colorbar = plt.colorbar(heatmap)
plt.show()

輸出

更改 Matplotlib 中顏色條的大小?

我們可以透過將值傳遞給 colorbar() 函式的 shrink 引數來更改顏色條的大小。如果需要減小顏色條的大小,shrink 值應小於 1。根據您傳遞的值,大小會減小。如果您將 0.5 作為值傳遞給此引數,則顏色條的大小將減半。

示例

這裡,我們使用 plot matplotlib 繪製了一個散點圖。對於該圖,我們建立了兩個列表“items”和“sales”,然後使用 plt.scatter() 語句建立了一個散點圖。我們建立了一個 shrink 值為 0.5 的顏色條。

import matplotlib.pyplot as plt
items=["chocolate","toffees","cake","colddrinks","lays","biscuit","frooti"]
sales=[89000,23000,98777,12000,65000,11100,96000]
c= ['r','y','m','r','y','m','b']
a= plt.scatter(items, sales, color=c)
plt.xlabel("ITEMS")
plt.ylabel("SALES")
plt.suptitle("SHOP MANGEMENT",fontsize=20)
plt.colorbar(a, shrink=0.5)
plt.show()

輸出

當 shrink 大小為 2 時,輸出如下 –

您也可以動態更改 Matplotlib 中顏色條的大小,編寫一個函式,該函式將以某種方式工作,即它將獲取顏色條的當前大小作為輸入值,並根據外部因素(例如資料量、繪圖大小)返回一個新大小。

結論

在本文中,我們討論了什麼是顏色條,它是如何建立的以及如何使用各種示例在 matplotlib 中更改顏色條的大小。

更新於: 2023-10-11

3K+ 次檢視

開啟您的 職業生涯

透過完成課程獲得認證

立即開始
廣告