python - 扩展 Pandas 数据框的简单/有效方法

Question

假设我有一个列表和一个 pandas 数据框

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

我想要一个大致如下所示的数据框:

a  b  y
0  0  1
1  1  1
0  0  2
1  1  2
0  0  3
1  1  3

有没有简单的方法可以做到这一点？

Answer 1

那叫cartesian product .转换为数据框

>>> y = pd.DataFrame(y, columns=list('y'))

添加常量键

>>> x['k'] = 1
>>> y['k'] = 1

并在其上合并

>>> pd.merge(y, x, on='k')[['a', 'b', 'y']]
   a  b  y
0  0  0  1
1  1  1  1
2  0  0  2
3  1  1  2
4  0  0  3
5  1  1  3