0

0

Pandas高效实现DataFrame笛卡尔积(交叉连接)

花韻仙語

花韻仙語

发布时间:2025-12-09 15:55:28

|

547人浏览过

|

来源于php中文网

原创

pandas高效实现dataframe笛卡尔积(交叉连接)

本教程详细介绍了如何使用Pandas库高效地在两个DataFrame之间执行笛卡尔积(交叉连接)操作,以生成所有可能的行组合。通过引入一个虚拟键并利用Pandas的`merge`函数,可以避免低效的循环,显著提高处理大型数据集时的性能,从而快速得到所需的行排列结果。

在数据分析和处理中,我们经常需要将两个或多个数据集中的所有可能元素进行组合,形成一个“全排列”或“笛卡尔积”(Cartesian Product)。例如,将一个日期列表与一个产品列表进行组合,以生成每个产品在每个日期的记录。当数据集较小时,使用循环结构或许可行,但对于大型Pandas DataFrame而言,基于循环的方法效率低下且资源消耗巨大。本教程将介绍一种在Pandas中实现笛卡尔积的高效、向量化方法,即通过引入一个虚拟键(dummy key)进行合并(merge)操作。

核心方法:利用虚拟键进行合并

Pandas的merge函数是处理DataFrame之间关系的核心工具。虽然它主要用于基于共同列进行内连接、左连接、右连接或外连接,但通过巧妙地引入一个共享的虚拟键,我们可以强制merge函数执行笛卡尔积。

基本原理:

  1. 为两个源DataFrame分别添加一个新列,该列包含相同的常量值(例如,整数1)。这个新列就是我们的“虚拟键”。
  2. 对这两个DataFrame执行内连接(inner merge),并指定以这个虚拟键作为连接依据。由于两个DataFrame的虚拟键列中的所有值都相同,merge函数会匹配所有可能的行对。
  3. 连接完成后,删除这个不再需要的虚拟键列,即可得到包含所有行组合的新DataFrame。

这种方法利用了Pandas底层的优化C/Cython实现,相比Python循环具有显著的性能优势。

示例代码

让我们使用提供的示例数据来演示这一过程。

原始DataFrame:

NewsBang
NewsBang

盛大旗下AI团队推出的智能新闻阅读App

下载
import pandas as pd

# DataFrame 1
data_1 = {'A1': ['2023-12-30', '2023-12-31']}
df_1 = pd.DataFrame(data_1)
print("df_1:")
print(df_1)

# DataFrame 2
data_2 = {'B1': ['Sam', 'Tam'],
          'B2': ['159cm', '175cm'],
          'B3': ['300gm', '400gm']}
df_2 = pd.DataFrame(data_2, index=[501, 502]) # 保持原始索引,虽然最终会被重置
print("\ndf_2:")
print(df_2)

输出:

df_1:
           A1
0  2023-12-30
1  2023-12-31

df_2:
     B1     B2     B3
501  Sam  159cm  300gm
502  Tam  175cm  400gm

执行笛卡尔积:

现在,我们将按照上述原理,为两个DataFrame添加虚拟键并进行合并。

# 1. 为两个DataFrame添加一个虚拟键列
df_1_temp = df_1.assign(key=1)
df_2_temp = df_2.assign(key=1)

print("\ndf_1_temp (带虚拟键):")
print(df_1_temp)
print("\ndf_2_temp (带虚拟键):")
print(df_2_temp)

# 2. 基于虚拟键进行内连接
df_result = pd.merge(df_1_temp, df_2_temp, on='key', how='inner')

# 3. 删除虚拟键列
df_result = df_result.drop('key', axis=1)

print("\ndf_result (笛卡尔积结果):")
print(df_result)

结果输出:

df_1_temp (带虚拟键):
           A1  key
0  2023-12-30    1
1  2023-12-31    1

df_2_temp (带虚拟键):
     B1     B2     B3  key
501  Sam  159cm  300gm    1
502  Tam  175cm  400gm    1

df_result (笛卡尔积结果):
           A1   B1     B2     B3
0  2023-12-30  Sam  159cm  300gm
1  2023-12-31  Sam  159cm  300gm
2  2023-12-30  Tam  175cm  300gm
3  2023-12-31  Tam  175cm  400gm

可以看到,df_result成功地生成了df_1的每一行与df_2的每一行的所有组合。

注意事项

  • 性能与内存: 这种方法对于中等大小的DataFrame非常高效。然而,笛卡尔积的结果DataFrame大小是两个原始DataFrame行数的乘积。如果原始DataFrame非常大(例如,都有数十万行),结果DataFrame可能会非常庞大,可能导致内存溢出(MemoryError)。在处理超大型数据集时,务必评估内存需求。
  • 列名冲突: 如果两个原始DataFrame中除了虚拟键之外还有同名的列,pd.merge会默认在冲突列名后添加_x和_y后缀。在实际应用中,您可能需要提前重命名列以避免歧义或在合并后进行处理。
  • 索引: 合并操作会重置DataFrame的索引。如果原始索引具有重要意义,您可能需要在合并前将其保存为普通列,或在合并后重新设置索引。
  • 代码简洁性: assign方法可以链式调用,使得代码更加简洁。例如:
    df_result_concise = (df_1.assign(key=1)
                         .merge(df_2.assign(key=1), on='key', how='inner')
                         .drop('key', axis=1))

总结

通过为两个DataFrame引入一个临时的虚拟键并利用Pandas的merge函数进行内连接,我们可以高效、简洁地实现笛卡尔积(交叉连接)操作。这种向量化的方法避免了低效的Python循环,是处理Pandas DataFrame全排列需求的推荐方式。尽管如此,在面对超大型数据集时,仍需警惕笛卡尔积可能带来的内存消耗问题。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

758

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

639

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

761

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

618

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1264

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

548

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

708

2023.08.11

高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

40

2026.01.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 2.3万人学习

Django 教程
Django 教程

共28课时 | 3.2万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号