python - 如何使用下面给出的 pandas 数据框在 python 中使用 seaborn 模块制作热图？

Question

这是一个国家数据框和每个国家拥有的汽车数量。最好将国家/地区放在左/y 轴，汽车作为底部/x 轴。

标签： pythonseaborn

Answer 1

只需将索引设置为country并通过sns.heatmap.
这是代码：

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

df=pd.DataFrame({'country':['us','france','spain','italy','germany'],
                 'corvette':[2,0,2,11,0],
                 'ford':[0,1,10,0,10],
                 'toyota':[1,10,0,1,1]})
df.set_index(['country'],inplace=True)
print(df) #1
ax=sns.heatmap(df,cmap='coolwarm')
plt.show() #2

输出：#1

         corvette  ford  toyota
country
us              2     0       1
france          0     1      10
spain           2    10       0
italy          11     0       1
germany         0    10       1

输出：#2