将DataFrame保存为同一个CSV文件
时间: 2023-11-22 07:55:31 浏览: 141
合并多个CSV文件与一个文件中
可以使用 pandas 库中的 to_csv() 方法将 DataFrame 保存为同一个 CSV 文件。其中,设置参数 mode='a' 可以在原有文件的末尾追加新的数据而不是覆盖原有数据。
示例代码如下:
```python
import pandas as pd
# 创建 DataFrame
data1 = {'name': ['Alice', 'Bob'], 'age': [25, 30]}
df1 = pd.DataFrame(data1)
data2 = {'name': ['Charlie', 'David'], 'age': [35, 40]}
df2 = pd.DataFrame(data2)
# 将两个 DataFrame 保存到同一个 CSV 文件
df1.to_csv('data.csv', mode='a', index=False, header=not bool(pd.read_csv('data.csv').shape[0]))
df2.to_csv('data.csv', mode='a', index=False, header=not bool(pd.read_csv('data.csv').shape[0]))
```
上述代码中,使用了 pandas.read_csv() 方法读取已有的 CSV 文件,判断是否需要写入表头。若文件为空,则需要写入表头;否则不需要写入。这样可以保证在同一个 CSV 文件中保存多个 DataFrame 时,只有第一个 DataFrame 会写入表头,后续的 DataFrame 不会重复写入。
阅读全文