将SAS代码转Python:实现DataFrame重复行合并方案咨询
SAS到Python:实现无关联条件的交叉连接(笛卡尔积)
嘿,我来帮你搞定这个转换问题!首先得拆解一下你给出的SAS代码逻辑:你写的proc sql里,on a.monthkey其实是利用了SAS的特性——当把数值变量放在on子句中时,SAS会将非零值判定为True。因为你的monthkey都是1-5的非零值,所以这个连接条件其实是恒成立的,最终会生成两个表的笛卡尔积(也就是每个monthkey和每个name都配对一次)。
在Python的pandas里,我们可以用以下两种方式实现同样的效果:
方法一:使用merge的cross连接(推荐,pandas 1.2.0+支持)
这是最直接的写法,专门用于生成两个表的交叉连接:
import pandas as pd df = pd.DataFrame({"monthkey": [1, 2, 3, 4, 5]}) df2 = pd.DataFrame({"name": ['foo','foo','bar']}) # 生成交叉连接(笛卡尔积) want = df.merge(df2, how='cross') print(want)
方法二:旧版本pandas的兼容写法
如果你的pandas版本低于1.2.0,不支持how='cross',可以通过添加临时辅助列来实现:
import pandas as pd df = pd.DataFrame({"monthkey": [1, 2, 3, 4, 5]}) df2 = pd.DataFrame({"name": ['foo','foo','bar']}) # 添加临时列,用于连接 df['tmp'] = 1 df2['tmp'] = 1 # 基于临时列做内连接,再删除临时列 want = df.merge(df2, on='tmp').drop('tmp', axis=1) print(want)
两种方法都会输出你想要的结果:
monthkey name 0 1 foo 1 1 foo 2 1 bar 3 2 foo 4 2 foo 5 2 bar 6 3 foo 7 3 foo 8 3 bar 9 4 foo 10 4 foo 11 4 bar 12 5 foo 13 5 foo 14 5 bar
内容的提问来源于stack exchange,提问作者Sasiwut Chaiyadecha
相关产品推荐
相关产品推荐

