首页 > 后端开发 > Python教程 > 如何高效计算 Pandas 中 DataFrame 的笛卡尔积?

如何高效计算 Pandas 中 DataFrame 的笛卡尔积?

Mary-Kate Olsen
发布: 2024-12-07 17:32:13
原创
542 人浏览过

How to Efficiently Calculate the Cartesian Product of DataFrames in Pandas?

Pandas 中的笛卡尔积

在 Pandas 中使用两个或多个数据帧时,可能需要获取它们的笛卡尔积,这会产生一个新的数据帧输入数据帧中行的所有组合。

Pandas >= 1.2

在 Pandas 的最新版本中,可以使用 merge 函数通过 how='cross' 参数来执行此操作。这种方法既简洁又高效:

import pandas as pd

df1 = pd.DataFrame({'col1':[1,2],'col2':[3,4]})
df2 = pd.DataFrame({'col3':[5,6]})

df_cartesian = df1.merge(df2, how='cross')
登录后复制

Pandas

1.2

对于早期版本的 Pandas,需要稍微不同的技术。它涉及创建一个为两个数据帧中的每一行重复的键列。添加此键列后,可以使用合并来执行笛卡尔积:
import pandas as pd
from pandas import merge

df1 = pd.DataFrame({'key':[1,1], 'col1':[1,2],'col2':[3,4]})
df2 = pd.DataFrame({'key':[1,1], 'col3':[5,6]})

merge(df1, df2,on='key')[['col1', 'col2', 'col3']]
登录后复制

这种方法涉及更多,但在旧版本的 Pandas 中有效。

以上是如何高效计算 Pandas 中 DataFrame 的笛卡尔积?的详细内容。更多信息请关注PHP中文网其他相关文章!

本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板