python:用pandas生成笛卡尔积

业务需要,把两组数据两两交叉组合,生成笛卡尔积,发现一个比较好用的方式,分享记录一下。

import pandas as pd
import numpy as np

data1 = {
    "a":[1,2],
    "b":[3,4]
}
data2 = {
    "c":[11,22],
    "d":[33,44],
    "e":[55,66]
}

df1 = pd.DataFrame(data1)
df2 = pd.DataFrame(data2)

df1['value']=1
df2['value']=1

df3 = df1.merge(df2,how='left',on='value')

del df3['value']

df3

最后结果:

        a	b	c	d	e
0	1	3	11	33	55
1	1	3	22	44	66
2	2	4	11	33	55
3	2	4	22	44	66

非常高效,有用~

posted @ 2020-12-26 10:56  水木青楓  阅读(5013)  评论(0编辑  收藏  举报