Jak tworzyć mapy cieplne za pomocą seaborn (z przykładami)

Mapa cieplna to rodzaj wykresu, który wykorzystuje różne odcienie kolorów do przedstawienia wartości danych.

W tym samouczku wyjaśniono, jak tworzyć mapy cieplne przy użyciu biblioteki wizualizacji Seaborn Python z następującym zbiorem danych:

 #import seaborn
import seaborn as sns

#load "flights" dataset
data = sns. load_dataset (“ flights ”)
data = data. pivot (" month ", " year ", " passengers ")

#view first five rows of dataset
data. head ()

Utwórz podstawową mapę cieplną

Możemy użyć następującej składni, aby utworzyć podstawową mapę cieplną dla tego zbioru danych:

 sns. heatmap (data)

Herbata

Dostosuj rozmiar mapy cieplnej

Możemy użyć argumentu figsize , aby dostosować ogólny rozmiar mapy cieplnej:

 #set heatmap size
import matplotlib. pyplot as plt
plt. figure (figsize = (12.8))

#create heatmap
sns. heatmap (data) 

Zmień kolory mapy cieplnej

Możemy użyć argumentu cmap , aby zmienić kolory używane w mapie cieplnej. Na przykład moglibyśmy wybrać paletę kolorów „Spectral”:

 sns. heatmap (data, cmap=" Spectral ")

Albo moglibyśmy wybrać „chłodną” kartę kolorów:

 sns. heatmap (data, cmap=" coolwarm ") 

Pełną listę dostępnych opcji cmap znajdziesz tutaj .

Opisz mapę cieplną

Możemy użyć następującej składni, aby opatrzyć każdą komórkę mapy cieplnej adnotacją w formacie liczb całkowitych i określić rozmiar czcionki:

 sns. heatmap (data, annot= True , fmt=" d ", annot_kws={" size ":13}) 

Zmień pasek kolorów mapy cieplnej

Na koniec możemy wyłączyć pasek kolorów, jeśli chcemy użyć argumentu cbar :

 sns. heatmap (data, cbar= False ) 

Więcej tutoriali Seaborn znajdziesz na tej stronie .