至于什么是相关系数,博主不在这里赘述了,大概能进来看的网友应该都知道,不会的可以上网找资料。直接上代码
# 餐饮销量数据相关性分析
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)
自己导入数据即可,改一下。
效果如下: