python实现相关系数的热力图绘制_sns 相关系数

至于什么是相关系数，博主不在这里赘述了，大概能进来看的网友应该都知道，不会的可以上网找资料。直接上代码

# 餐饮销量数据相关性分析
import pandas as pd
catering_sale = 'E:\数据分析与数据挖掘数据集\《Python数据分析与挖掘实战（第2版）》源数据和代码-各章节\chapter3\demo\data\catering_sale_all.xls'
data = pd.read_excel(catering_sale, index_col='日期')  # 读取数据，指定日期列作为索引列
# import warnings
# warnings.filterwarnings("ignore")
# print(data.corr())  # 计算相关系数矩阵
import matplotlib.pyplot as plt
# import seaborn as sns
import numpy as np
import seaborn as sns
# import matplotlib.pyplot as plt
def test(df):
    dfData = df.corr()
    plt.rcParams['font.sans-serif'] = ['SimHei']  # 正常显示中文
    plt.rcParams['axes.unicode_minus'] = False
    plt.subplots(figsize=(9, 9)) # 设置画面大小
    sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues")
    plt.savefig('./相关系数热力图.png')
    plt.show()
test(data)

自己导入数据即可，改一下。

效果如下：