计算两个DataFrame间Cohen's Kappa相似度矩阵

本教程详细介绍了如何使用Python和Pandas库计算两个DataFrame中列表数据的Cohen's Kappa系数，并将其组织成一个完整的相似度矩阵。文章涵盖了从数据准备、核心计算逻辑到构建跨DataFrame和内部DataFrame的综合相似度矩阵的方法，旨在帮助读者高效地评估不同主体间的一致性。

引言：Cohen's Kappa与数据一致性评估

Cohen's Kappa系数是一种衡量两个评估者之间一致性的统计量，它在考虑了偶然一致性之后，评估分类数据的一致程度。其值通常介于-1到1之间，其中1表示完全一致，0表示偶然一致，负值表示一致性低于偶然水平。在处理多主体或多组数据时，我们常常需要计算不同主体之间的配对一致性，例如，评估不同实验组或不同数据集中的个体行为模式是否一致。

本教程将演示如何处理两个独立的Pandas DataFrame，它们包含各自的主体（subject）和对应的分类列表数据（lists），并计算所有主体之间的Cohen's Kappa配对分数，最终将结果组织成一个统一的相似度矩阵。

数据准备与环境配置

首先，我们需要导入必要的库并准备示例数据。我们将使用pandas进行数据操作，以及sklearn.metrics中的cohen_kappa_score函数来计算Kappa值。

import pandas as pd
from sklearn.metrics import cohen_kappa_score

# 示例数据框 df1
data1 = {'subject': ['A', 'B', 'C', 'D'],
         'group': ['red', 'red', 'blue', 'blue'],
         'lists': [[0, 1, 1], [0, 0, 0], [1, 1, 1], [0, 1, 0]]}
df1 = pd.DataFrame(data1)

# 示例数据框 df2
data2 = {'subject': ['a', 'b', 'c', 'd'],
         'group': ['red', 'red', 'blue', 'blue'],
         'lists': [[0, 1, 0], [1, 1, 0], [1, 0, 1], [1, 1, 0]]}
df2 = pd.DataFrame(data2)

print("DataFrame 1:")
print(df1)
print("\nDataFrame 2:")
print(df2)

在这个例子中，df1和df2分别代表了两组不同的主体及其对应的分类列表。我们的目标是计算df1中的每个主体与df2中的每个主体之间的Kappa值，并最终构建一个包含所有主体（包括df1和df2中的）的全面相似度矩阵。

核心任务：计算DataFrame间的配对Kappa值

首先，我们来看如何计算df1中的每个主体与df2中的每个主体之间的Kappa值，并将其组织成一个DataFrame。这可以通过嵌套的字典推导式实现，它能高效地遍历两个DataFrame的行。

# 计算 df1 主体与 df2 主体之间的配对 Kappa 值
kappa_df1_vs_df2 = pd.DataFrame({
    i.subject: {
        j.subject: cohen_kappa_score(i.lists, j.lists)
        for j in df2.itertuples()
    }
    for i in df1.itertuples()
})

print("\nKappa Scores (df1 subjects vs df2 subjects):")
print(kappa_df1_vs_df2)

这段代码利用itertuples()方法高效地遍历DataFrame的行，并以命名元组的形式访问行数据（如i.subject和i.lists）。外层循环遍历df1的每一行，内层循环遍历df2的每一行，计算Kappa值并构建一个嵌套字典，最终转换为一个Pandas DataFrame。这个DataFrame的行索引是df1的主体，列索引是df2的主体。

构建全面的相似度矩阵（跨DataFrame与内部DataFrame）

为了实现用户最终目标——一个包含所有主体（来自df1和df2）作为行和列的完整相似度矩阵，我们需要一个更通用的方法。这包括计算df1内部主体之间、df2内部主体之间以及df1和df2主体之间的所有配对Kappa值。

乾坤圈新媒体矩阵管家

新媒体账号、门店矩阵智能管理系统

17

查看详情

最简洁的方法是首先将两个DataFrame中所有主体的数据合并到一个统一的数据结构中，然后对这个统一结构中的所有主体进行两两比较。

# 1. 合并所有主体的列表数据
# 使用set_index('subject')以便通过主体名称快速查找数据
all_subject_data = pd.concat([df1.set_index('subject'), df2.set_index('subject')])

# 获取所有主体的唯一列表，用于构建矩阵的索引和列名
all_subjects = list(all_subject_data.index)

# 2. 初始化一个空的DataFrame作为Kappa矩阵
# 使用dtype=float确保数值类型
kappa_matrix = pd.DataFrame(index=all_subjects, columns=all_subjects, dtype=float)

# 3. 遍历所有主体对，计算Kappa值并填充矩阵
for sub1 in all_subjects:
    for sub2 in all_subjects:
        list1 = all_subject_data.loc[sub1, 'lists']
        list2 = all_subject_data.loc[sub2, 'lists']
        kappa_matrix.loc[sub1, sub2] = cohen_kappa_score(list1, list2)

print("\nComprehensive Pairwise Kappa Similarity Matrix:")
print(kappa_matrix)

这个方法首先通过pd.concat将df1和df2合并，并以subject列作为索引，这样可以方便地通过主体名称loc定位到其对应的lists数据。然后，我们创建一个空的kappa_matrix，其行和列都是所有主体的列表。最后，通过双重循环遍历所有可能的主体对，计算它们的cohen_kappa_score并填充到矩阵中。

结果解读与可视化提示

生成的kappa_matrix是一个对称矩阵（因为cohen_kappa_score(A, B)等于cohen_kappa_score(B, A)），其对角线上的值通常为1（一个主体与自身比较）。这个矩阵清晰地展示了所有主体之间的相似性。

例如，在上面的输出中：

• kappa_matrix.loc['A', 'a']显示了df1中主体'A'与df2中主体'a'之间的一致性。
• kappa_matrix.loc['A', 'B']显示了df1中主体'A'与df1中主体'B'之间的一致性。
• kappa_matrix.loc['a', 'b']显示了df2中主体'a'与df2中主体'b'之间的一致性。

这个矩阵是进行进一步分析的基础，例如，可以通过seaborn.heatmap将其可视化，以便直观地观察不同组或不同主体之间的一致性模式。用户可以根据group信息对矩阵的行和列进行排序，从而更好地识别“组内”和“组间”的一致性差异。

注意事项

• 数据类型： cohen_kappa_score函数期望输入为一维数组或列表。确保lists列中的数据是适合这种格式的。
• 计算效率： 对于非常大的数据集（数千个主体），这种双重循环的方法计算复杂度为O(N^2)，可能会非常耗时。在这种情况下，可能需要考虑使用更优化的并行计算或近似方法。
• Kappa值解释： Kappa值的大小需要结合具体领域知识进行解释。通常，0.21-0.40被认为是"一般"一致性，0.41-0.60为"中等"，0.61-0.80为"良好"，0.81-1.00为"非常好"。负值和接近0的值表明一致性很差或仅为偶然。

总结

本教程提供了一种在Python中计算两个DataFrame之间以及内部所有主体配对Cohen's Kappa相似度的方法。通过合并数据并构建一个统一的相似度矩阵，我们能够全面地评估不同主体间的一致性，为后续的数据分析和可视化（如热力图）奠定基础。这种方法不仅灵活，而且易于理解和实现，是处理分类数据一致性评估的有效工具。

以上就是计算两个DataFrame间Cohen's Kappa相似度矩阵的详细内容，更多请关注php中文网其它相关文章！