如何在Pandas中高效生成多行元素两两组合并聚合统计值

来源:个人站长作者:菲律宾程序员头衔:程序员
导读:本期聚焦于小伙伴创作的《如何在Pandas中高效生成多行元素两两组合并聚合统计值》,敬请观看详情,探索知识的价值。以下视频、文章将为您系统阐述其核心内容与价值。如果您觉得《如何在Pandas中高效生成多行元素两两组合并聚合统计值》有用,将其分享出去将是对创作者最好的鼓励。

在数据分析工作中,我们常需要把一张表里的多行记录按两两配对的方式生成组合,再针对每对组合做计数、求和等聚合操作。如果数据量较大,用Python双层循环会非常慢。下面介绍几种在Pandas里高效实现该需求的方法。

如何在Pandas中高效生成多行元素两两组合并聚合统计值

方法一:自连接与groupby聚合

假设有一张表记录用户每天的点击商品,我们想统计任意两个商品被同一用户点击的配对次数。可以先给原表取别名做自连接,再过滤掉自身配对和重复无序对,最后统计。

import pandas as pd

# 构造示例数据
df = pd.DataFrame({
    'user': ['u1', 'u1', 'u2', 'u2', 'u3'],
    'item': ['a', 'b', 'a', 'c', 'b']
})

# 自连接生成两两组合
merged = df.merge(df, on='user')
# 只保留 item_x < item_y 的无序对,避免重复和自身配对
pair = merged[merged['item_x'] < merged['item_y']]
# 聚合统计每对商品出现次数
result = pair.groupby(['item_x', 'item_y']).size().reset_index(name='count')
print(result)

方法二:使用itertools.combinations

当每个分组内元素不多、分组数量大时,可以先按分组拿出元素列表,再用combinations展开,最后统一聚合。

from itertools import combinations
import pandas as pd

df = pd.DataFrame({
    'user': ['u1', 'u1', 'u2', 'u2'],
    'item': ['a', 'b', 'a', 'c']
})

rows = []
for user, group in df.groupby('user'):
    for x, y in combinations(sorted(group['item']), 2):
        rows.append((x, y))
pairs = pd.DataFrame(rows, columns=['item_x', 'item_y'])
result = pairs.groupby(['item_x', 'item_y']).size().reset_index(name='count')
print(result)

方法三:矩阵化提升性能

如果关注全局元素共现,可先建透视表得到用户-物品矩阵,再用点积算共现矩阵,取上三角避免重复。

import pandas as pd
import numpy as np

df = pd.DataFrame({
    'user': ['u1', 'u1', 'u2', 'u2', 'u3'],
    'item': ['a', 'b', 'a', 'c', 'b']
})

mat = pd.crosstab(df['user'], df['item'])
cooc = mat.T.dot(mat)
# 将对角线设为0,并取上三角
np.fill_diagonal(cooc.values, 0)
upper = cooc.where(np.triu(np.ones(cooc.shape), 1).astype(bool))
print(upper)

总结

对于Pandas中多行元素两两组合并聚合统计值,小数据可用自连接或combinations,大数据建议矩阵运算。合理选择方法能兼顾代码可读性与执行效率。

Pandas两两组合聚合统计修改时间:2026-07-27 08:09:18

免责声明:​ 已尽一切努力确保本网站所含信息的准确性。网站内容多为原创整理与精心编撰,观点力求客观中立。本站旨在免费分享,内容仅供个人学习、研究或参考使用。若引用了第三方作品,版权归原作者所有。如内容涉及您的权益,请联系我们处理。
内容垂直聚焦
专注技术核心技术栏目,确保每篇文章深度聚焦于实用技能。从代码技巧到架构设计,为用户提供无干扰的纯技术知识沉淀,精准满足专业提升需求。
知识结构清晰
覆盖从开发到部署的全链路。AI、前端、编程、数据库、服务器、建站、系统层层递进,构建清晰学习路径,帮助用户系统化掌握开发与运维所需的核心技术。
深度技术解析
拒绝泛泛而谈,深入技术细节与实践难点。无论是数据库优化还是服务器配置,均结合真实场景与代码示例进行剖析,致力于提供可直接应用于工作的解决方案。
专业领域覆盖
精准对应开发生命周期。从前端界面到后端编程,从数据库操作到服务器运维,形成完整闭环,一站式满足全栈工程师和运维人员的技术需求。
即学即用高效
内容强调实操性,步骤清晰、代码完整。用户可根据教程直接复现和应用于自身项目,显著缩短从学习到实践的距离,快速解决开发中的具体问题。
持续更新保障
专注既定技术方向进行长期、稳定的内容输出。确保各栏目技术文章持续更新迭代,紧跟主流技术发展趋势,为用户提供经久不衰的学习价值。