python - 合并两个数据帧以模仿 'select * from tablea, tableb'

Question

这个问题已经有答案了:

cartesian product in pandas (13 个回答)

已关闭 4 年前。

我想将两个数据帧合并在一起以模仿“select * from A, B”，例如，表 A 将如下所示:

import pandas as pd
table_a = pd.DataFrame({'id': ['A', 'B']})

表 B 将如下所示:

import pandas as pd
table_b = pd.DataFrame({'x': [1,2], 'y':[3,4]})

我想要的是

import pandas as pd
table_c = pd.DataFrame({'id':['A', 'A', 'B', 'B'],
                        'x': [1,2,1,2], 'y':[3,4,3,4]})

即复制 A 和 B table_b.shape[0] 次并追加。

我相信有多种方法可以做到这一点。例如，

a = table_b.loc[np.tile(table_b.index.values, table_a.shape[0])]
b = table_a.loc[np.repeat(table_a.index.values, table_b.shape[0])]
a['id'] = b['id'].values

还有其他更好的方法吗？谢谢

Answer 1

pd.DataFrame([
    d1 + d2
    for d1 in zip(*map(table_a.get, table_a))
    for d2 in zip(*map(table_b.get, table_b))
], columns=table_a.columns.append(table_b.columns))