Seaborn热图按列

Question

在我看来,热图功能完全应用于数据帧.如果我只想将热图应用于我的数据集中给定的一组列,该怎么办？我想这可以通过巧妙地使用cmap来实现,但似乎无法让它工作.

Answer 1

将所需的子DataFrame传递给seaborn.heatmap:

seaborn.heatmap(df[[col1, col2]], ...)

df[[col1, col2, ..., coln]]返回列组成的数据帧col1,col2... coln从df.请注意双括号.

如果您只想突出显示某些值并绘制热图,就像所有其他值都为零一样,您可以复制DataFrame并在调用之前将这些值设置为零heatmap.例如,从文档修改示例,

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import seaborn.matrix as smatrix

sns.set()

flights_long = sns.load_dataset("flights")
flights = flights_long.pivot("month", "year", "passengers")
flights = flights.reindex(flights_long.iloc[:12].month)

columns = [1953,1955]
myflights = flights.copy()
mask = myflights.columns.isin(columns)
myflights.loc[:, ~mask] = 0
arr = flights.values
vmin, vmax = arr.min(), arr.max()
sns.heatmap(myflights, annot=True, fmt="d", vmin=vmin, vmax=vmax)
plt.show()

产量