怎么用pandas库在csv文件中定义新的两列并写入数据
时间: 2023-05-26 12:02:14 浏览: 161
基于pandas向csv添加新的行和列
可以使用pandas的DataFrame对象来处理CSV文件。下面是一个示例代码,该代码可以在CSV文件中定义新的两列,并将数据写入其中:
```python
import pandas as pd
# 读入csv文件,假设文件名为data.csv
df = pd.read_csv('data.csv')
# 在DataFrame对象中定义新列
df['new_col1'] = [1, 2, 3] # 通过列表的方式定义列
df['new_col2'] = pd.Series([4,5,6], index=df.index) # 通过Series对象的方式定义列
# 写入csv文件,假设输出文件名为output.csv
df.to_csv('output.csv', index=False)
```
在上面的示例中,在CSV文件中定义了两列`new_col1`和`new_col2`,并将它们的值赋为`[1, 2, 3]`和`[4, 5, 6]`。随后,调用`to_csv()`方法将修改过的DataFrame对象写入到输出文件`output.csv`中。其中,`index=False`表示不保存DataFrame对象的行索引到输出文件中。
阅读全文