怎么用pandas库在csv文件中定义新的一列并写入数据
时间: 2023-05-26 15:01:58 浏览: 100
基于pandas向csv添加新的行和列
可以使用 pandas 库的 `read_csv()` 方法读取旧 CSV 文件,然后使用 `DataFrame` 对象的 `assign()` 方法添加新列并写入新的数据,最后使用 `to_csv()` 方法将新的 `DataFrame` 对象写入到新的 CSV 文件中。
以下是一个示例代码:
```python
import pandas as pd
# 读取旧 CSV 文件
old_df = pd.read_csv('old.csv')
# 添加新列并写入数据
new_df = old_df.assign(new_column=[1, 2, 3])
# 将新 DataFrame 对象写入到新的 CSV 文件中
new_df.to_csv('new.csv', index=False)
```
在上面的示例代码中,`assign()` 方法接收一个新列名和一个列表作为参数,将新列名作为 DataFrame 对象的列名并将列表中的数据填充到新列中。`to_csv()` 方法接收一个新文件名并使用给定参数将新 DataFrame 对象写入到新的 CSV 文件中。在这个示例中,`index=False` 表示不将行索引写入到 CSV 文件中。
阅读全文