溫馨提示×

Seaborn中的熱圖功能怎么使用

小億
84
2024-05-17 11:09:15
欄目: 編程語言

要使用Seaborn中的熱圖功能，首先需要導(dǎo)入Seaborn庫并載入數(shù)據(jù)集。然后使用Seaborn的heatmap()函數(shù)來繪制熱圖。

下面是一個(gè)簡單的示例代碼，演示如何使用Seaborn繪制熱圖：

import seaborn as sns
import matplotlib.pyplot as plt

# 載入數(shù)據(jù)集
data = sns.load_dataset("flights")

# 通過數(shù)據(jù)集創(chuàng)建一個(gè)矩陣
data_pivot = data.pivot("month", "year", "passengers")

# 繪制熱圖
sns.heatmap(data_pivot, annot=True, fmt="d", cmap="YlGnBu")

# 添加標(biāo)題
plt.title("Passengers Heatmap")

# 顯示圖形
plt.show()

在上面的示例中，我們首先載入了Seaborn庫和一個(gè)名為"flights"的示例數(shù)據(jù)集。然后，我們使用pivot()函數(shù)創(chuàng)建了一個(gè)適合繪制熱圖的矩陣。最后，我們使用heatmap()函數(shù)繪制了熱圖，并通過設(shè)置參數(shù)annot為True來在每個(gè)方塊上顯示數(shù)據(jù)標(biāo)簽。我們還設(shè)置了顏色映射為"YlGnBu"。

您可以根據(jù)具體的數(shù)據(jù)和需求進(jìn)行定制化，例如調(diào)整顏色映射、添加行列標(biāo)簽、設(shè)置標(biāo)簽格式等。希望這個(gè)示例能幫助您開始使用Seaborn繪制熱圖。

0 踩