当前位置: 代码网 > it编程>前端脚本>Python > Spearman 相关性分析法,以及python的完整代码应用

Spearman 相关性分析法,以及python的完整代码应用

2024年08月01日 Python 我要评论
Spearman 相关性分析法是一种针对两个变量之间非线性关系的相关性计算方法,同时,它不对数据的分布进行假设。
  • spearman 相关性分析法

简介

spearman 相关性分析法是一种针对两个变量之间非线性关系的相关性计算方法,同时,它不对数据的分布进行假设。该方法的基本思想是将两个(也可以多个)变量的值进行排序,并计算它们之间的等级相关性(spearman 相关系数)。spearman 相关系数的范围在 -1 到 1 之间,取值为 -1 表示完全负相关,取值为 1 表示完全正相关,取值为 0 则表示两个变量之间没有相关性。

基础使用方法

代码示例

import pandas as pd
import seaborn as sns

# 构造数据
data = {
    'x': [1, 3, 5, 7, 9],
    'y': [10, 8, 6, 4, 2]
}
df = pd.dataframe(data)

# spearman 相关性分析
corr = df.corr(method='spearman')
print(corr)

# 绘制相关系数热力图
sns.heatmap(corr, annot=true, cmap="ylgnbu")

参数说明

method: 相关性分析方法,这里需要指定为 spearman

函数返回值

sns.heatmap(corr, annot=true, cmap="ylgnbu") 会绘制出相关系数热力图,其中:

  • corr: 相关系数,即 spearman 相关系数。
  • annot: 是否在每个方格中显示数值。
  • cmap: 颜色映射表。

引入多个变量

  • spearman 相关性分析法通常用于分析两个变量之间的关系,但也可以通过计算多组变量之间的 spearman 相关系数来分析它们之间的相互关系。对于多个变量之间的相关性分析,可以使用 pandas 中的 corr 函数计算这些变量之间的 spearman 相关系数矩阵。然后,可以使用 seaborn 中的 heatmap 函数绘制相关系数矩阵的热力图。
import pandas as pd
import seaborn as sns

# 构造数据
data = {
    'x1': [1, 3, 5, 7, 9],
    'x2': [10, 8, 6, 4, 2],
    'x3': [9, -7, 5.4, -3, 1],
    'x4': [2, 4, 6, 8, 10]
}
df = pd.dataframe(data)

# spearman 相关性分析
corr = df.corr(method='spearman')
print(corr)

# 绘制相关系数热力图
sns.heatmap(corr, annot=true, cmap="ylgnbu")

在上面的示例代码中,我们构造了一个包含了 4 个变量的数据集,包括 ‘x1’、‘x2’、‘x3’ 和 ‘x4’,然后使用 spearman 相关性分析法计算了这些变量之间的 spearman 相关系数矩阵,并绘制了相关系数的热力图。
结果图

生成热力图的保存

  • 要将生成的 python 热力图保存为文件,需要使用 matplotlib 库的 savefig() 函数:
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

# 构造数据(可以使用你的数据)
data = {
    'x1': [1, 3, 5, 7, 9],
    'x2': [10, 8, 6, 4, 2],
    'x3': [9, -7, 5.4, -3, 1],
    'x4': [2, 4, 6, 8, 10]
}
df = pd.dataframe(data)

# spearman 相关性分析
corr = df.corr(method='spearman')
print(corr)

# 绘制相关系数热力图
sns.heatmap(corr, annot=true, cmap="ylgnbu")

# 保存热力图
plt.savefig('heatmap.png', dpi=300, bbox_inches='tight')

在这个示例代码中,使用了 matplotlib 的 savefig() 函数来保存热力图。这个函数的第一个参数是文件的名称和路径。在这个示例中,我们指定了 "heatmap.png" 作为文件名,并将热力图保存在当前工作目录下。dpi 参数设定每英寸像素点数(dots per inch),默认为 100,一般可以设定为 300,以获得更高的分辨率。bbox_inches='tight' 参数用来紧缩图像区域,以避免被裁剪,可以根据需要进行调整。

从excel读取数据

从 excel 表中获取各个变量的数据,可以使用 python 中的 pandas 库。pandas 可以读取 excel 表中的数据,并将其转换为 pandas dataframe 对象,以便进行数据分析和可视化。

示例代码:

import pandas as pd
import seaborn as sns

# 读取 excel 表中的数据
df = pd.read_excel('data.xlsx')

# spearman 相关性分析
corr = df.corr(method='spearman')
print(corr)

# 绘制相关系数热力图
sns.heatmap(corr, annot=true, cmap="ylgnbu")

其中,read_excel() 函数用于读取 excel 文件,其参数指定 excel 文件的名称和路径。默认情况下,它读取文件中的第一个工作表,并将其转换为 pandas dataframe 对象。如果 excel 文件中有多个工作表,可以使用 sheet_name 参数将指定的工作表读取为 dataframe 对象。

假设你的 excel 文件中每一列数据代表一个变量,可以使用以下代码将 excel 表中的每列数据分离成新的 dataframe 对象:

# 将列分离成新的 dataframe 对象
var_dict = {}
for column in df:
    var_dict[column] = df[column]

# 输出各个变量的数据
for key, value in var_dict.items():
    print(key, value.tolist())

df[column] 返回一个 pandas series 对象,其中包含 excel 表中某一列的数据。将其存储在一个 python 的字典对象 var_dict 中,可以使用 tolist() 函数将其转换为 python 列表以输出各个变量的数据。

然后,可以使用上面的示例代码,在所述的每一列中提取来自幻灯片的各个变量数据,并使用 spearman 相关性分析法计算它们之间的关系,最后生成相关系数热力图。
较复杂的热力图

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com