Python相关性分析

Roy Teng2024-04-22 1:36

分析连续变量之间线性相关程度的强弱，并用适当的统计指标表示出来的过程称为相关分析。

可以直接绘制散点图，或者绘制散点图矩阵，或者计算相关系数来进行相关分析。

相关系数的计算如下所示：

示例数据：

计算百合酱蒸凤爪与其他几种菜品的相关系数代码如下：

python 复制代码

import pandas as pd
catering_sale = r'D:\daily\data\catering_sale_all.xls' #餐饮数据，含有其他属性
data = pd.read_excel(catering_sale, index_col = u'日期') #读取数据，指定"日期"列为索引列
data.corr() #相关系数矩阵
data.corr()[u'百合酱蒸凤爪'] #"百合酱蒸凤爪"与其他菜式的相关系数
data[u'百合酱蒸凤爪'].corr(data[u'翡翠蒸香茜饺']) #计算"百合酱蒸凤爪"与"翡翠蒸香茜饺"的相关系数
# 0.009205803051836475

运行结果如下：

相关系数矩阵：

根据相关系数矩阵绘制热力图

运行代码：

python 复制代码

import seaborn as sns
import matplotlib.pyplot as plt
# 生成相关系数矩阵
corr = data.corr()
# 绘制热力图
plt.figure(figsize=(10, 8))
sns.heatmap(corr, annot=True, cmap='coolwarm', fmt=".2f")
plt.title('Correlation Heatmap')
plt.show()

运行结果如下所示：