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)
自己导入数据即可,改一下。
效果如下:
