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

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

# 餐饮销量数据相关性分析
import pandas as pd
catering_sale = E:数据分析与数据挖掘数据集《Python数据分析与挖掘实战(第2版)》源数据和代码-各章节chapter3demodatacatering_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)

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

效果如下:

经验分享 程序员 微信小程序 职场和发展