python csv 写文件
时间: 2024-12-26 15:14:21 浏览: 1
### 使用Python将数据写入CSV文件
为了实现这一目标,可以利用`csv`库中的功能来处理CSV文件。下面是一个简单的例子,展示了如何创建一个新的CSV文件并将一些数据写入其中。
```python
import csv
# 定义要保存到CSV的数据列表
data = [
['姓名', '年龄', '城市'],
['张三', 30, '北京'],
['李四', 25, '上海']
]
with open('output.csv', mode='w', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
# 将每一行数据写入CSV文件
for row in data:
writer.writerow(row)
```
这段代码首先导入了必要的模块,并定义了一个包含多条记录的二维数组作为待写入的内容。接着通过`open()`函数打开指定路径下的文件(如果不存在则会自动创建),设置模式为只写(`'w'`)并指定了编码方式防止中文乱码问题[^1]。最后使用`writerow()`方法逐行向CSV文件中添加新纪录。
对于更复杂的情况,比如当需要处理大量结构化表格型数据集时,推荐采用Pandas库来进行操作。它提供了更加便捷高效的API接口用于读取/修改以及导出DataFrame对象至CSV格式:
```python
import pandas as pd
df = pd.DataFrame({
'姓名': ['王五', '赵六'],
'年龄': [27, 33],
'城市': ['广州', '深圳']
})
# 导出DataFrame到CSV文件
df.to_csv('people_info.csv', index=False, encoding='utf_8_sig')
```
这里先构建了一个名为`df`的DataFrame实例,包含了若干列名及其对应的值序列;之后调用了`.to_csv()`方法将其转换成CSV文档形式存储于本地磁盘上。参数`index=False`表示不保留原始索引信息而仅输出实际业务字段;另外还特别设置了字符编码为UTF-8带BOM签名的形式以兼容某些特定环境下的文本解析需求[^2]。
阅读全文