如何在 python 中創建 dataframe 的熱圖

Question

我正在嘗試創建熱圖並關注以下問題：

從 pandas DataFrame 制作熱圖

我的 dataframe 如下圖所示：

我嘗試了以下代碼：

years = ["1860","1870", "1880","1890","1900","1910","1920","1930","1940","1950","1960","1970","1980","1990","2000"]
kantons = ["AG","AI","AR","BE","BL","BS","FR","GE","GL","GR","JU","LU","NE","NW","OW","SG","SH","SO","SZ","TG","TI","UR","VD","VS","ZG","ZH"]
   
df = pd(abs(dfYears), index=years, columns=kantons)

這給出了“AG”不能用作浮點數的例外。 所以我想如果我需要刪除不可能的索引列。

有什么建議么？

Answer 1

復制類似數據時，您可以執行以下操作：

import pandas as pd
import numpy as np

years = ["1860","1870", "1880","1890","1900","1910","1920","1930","1940","1950","1960","1970","1980","1990","2000"]
kantons = ["AG","AI","AR","BE","BL","BS","FR","GE","GL","GR","JU","LU","NE","NW","OW","SG","SH","SO","SZ","TG","TI","UR","VD","VS","ZG","ZH"]
   
df = pd.DataFrame(np.random.randint(low=10000, high=200000, size=(15, 26)), index=years, columns=kantons)
df.style.background_gradient(cmap='Reds')

Pandas 有一些內置 Styles用於最常見的可視化需求。 .background_gradient function 是一種基於單元格值突出顯示單元格的簡單方法。 cmap參數根據matplotlib 顏色圖確定顏色 map。