python合并dataframe后只保存几列
时间: 2023-07-06 15:35:53 浏览: 81
python之DataFrame实现excel合并单元格
5星 · 资源好评率100%
您可以在合并DataFrame之后,使用`.loc`方法选择您想要的列,并将其保存到新的DataFrame中。举个例子,假设您已经将两个DataFrame `df1`和`df2`按照某个共同的列进行了合并,并想要保存其中的列`col1`和`col2`,可以按照以下方式实现:
```python
new_df = pd.merge(df1, df2, on='common_col')[['col1', 'col2']]
```
这里我们选择了在合并后的DataFrame中选取`col1`和`col2`这两列,并将它们放入一个新的DataFrame `new_df`中。
如果您想要保留其他列,也可以在列表中将它们包含在内,例如:
```python
new_df = pd.merge(df1, df2, on='common_col')[['col1', 'col2', 'col3', 'col4']]
```
这里我们选择了`col1`、`col2`、`col3`和`col4`这四列,并将它们放入了`new_df`中。
阅读全文