python相关性矩阵热图_python – 相关热图

另一种方法是使用seaborn中的热图函数来绘制协方差.此示例使用R中ISLR程序包中的自动数据集(与您显示的示例中的相同).

import pandas.rpy.common as com

import seaborn as sns

%matplotlib inline

# load the R package ISLR

infert = com.importr("ISLR")

# load the Auto dataset

auto_df = com.load_data('Auto')

# calculate the correlation matrix

corr = auto_df.corr()

# plot the heatmap

sns.heatmap(corr,

xticklabels=corr.columns,

yticklabels=corr.columns)

如果你想更加喜欢,你可以使用Pandas Style,例如：

cmap = cmap=sns.diverging_palette(5, 250, as_cmap=True)

def magnify():

return [dict(selector="th",

props=[("font-size", "7pt")]),

dict(selector="td",

props=[('padding', "0em 0em")]),

dict(selector="th:hover",

props=[("font-size", "12pt")]),

dict(selector="tr:hover td:hover",

props=[('max-width', '200px'),

('font-size', '12pt')])

]

corr.style.background_gradient(cmap, axis=1)\n .set_properties(**{'max-width': '80px', 'font-size': '10pt'})\n .set_caption("Hover to magify")\n .set_precision(2)\n .set_table_styles(magnify())