python - 添加从两个 pandas dfs 中选择的列

Question

我有两个 pandas 数据框 a_df 和 b_df。 a_df 具有 ID、atext 和 var1-var25 列，而 b_df 具有 ID、atext 和 var1-var 25 列。

我只想添加 a_df 和 b_df 中的相应变量，并保留 ID 和 atext。

下面的代码添加所有相应的列。有没有办法让它只添加感兴趣的列？

absum_df=a_df.add(b_df)

我可以做什么来实现这个目标？

Answer 1

使用 filter :

absum_df = a_df.filter(like='var').add(b_df.filter(like='var'))

如果您想按原样保留其他列，请使用 concat求和后:

absum_df = pd.concat([a_df[['ID', 'atext']], absum_df], axis=1)

或者，不要从 a_df 中子选择列，您可以直接将 absum_df 中的列删除，如果您想添加 a_df 中的所有列不在absum_df中:

absum_df = pd.concat([a_df.drop(absum_df.columns axis=1), absum_df], axis=1)