可以使用pandas
库中的merge
函数来按组合并和扩展数据框。下面是一个示例代码:
import pandas as pd
# 创建两个数据框
df1 = pd.DataFrame({'A': ['A0', 'A1', 'A2', 'A3'],
'B': ['B0', 'B1', 'B2', 'B3'],
'key': ['K0', 'K1', 'K2', 'K3']})
df2 = pd.DataFrame({'C': ['C0', 'C1', 'C2', 'C3'],
'D': ['D0', 'D1', 'D2', 'D3'],
'key': ['K0', 'K1', 'K2', 'K3']})
# 按照 'key' 列合并两个数据框
merged_df = pd.merge(df1, df2, on='key')
# 打印合并后的结果
print(merged_df)
运行以上代码,将得到合并后的结果:
A B key C D
0 A0 B0 K0 C0 D0
1 A1 B1 K1 C1 D1
2 A2 B2 K2 C2 D2
3 A3 B3 K3 C3 D3
上述代码中,我们创建了两个数据框df1
和df2
,然后使用merge
函数按照'key'列将两个数据框合并成一个新的数据框merged_df
。最后,我们打印出合并后的结果。