导出CSV模块
时间: 2023-07-13 07:07:31 浏览: 43
Python中可以使用`csv`模块来导出CSV文件。以下是一个简单的示例:
```python
import csv
data = [
['Name', 'Age', 'Country'],
['John', 25, 'USA'],
['Mary', 30, 'Canada'],
['Bob', 35, 'Australia']
]
with open('output.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(data)
```
以上代码将一个二维数组`data`写入到名为`output.csv`的文件中。`newline=''`参数用于确保在Windows操作系统上正确地写入换行符。如果不加这个参数,将会出现空行。
如果要导出的数据是字典形式,可以使用`DictWriter`类。以下是一个示例:
```python
import csv
data = [
{'Name': 'John', 'Age': 25, 'Country': 'USA'},
{'Name': 'Mary', 'Age': 30, 'Country': 'Canada'},
{'Name': 'Bob', 'Age': 35, 'Country': 'Australia'}
]
with open('output.csv', 'w', newline='') as f:
fieldnames = ['Name', 'Age', 'Country']
writer = csv.DictWriter(f, fieldnames=fieldnames)
writer.writeheader()
writer.writerows(data)
```
以上代码将一个字典列表`data`写入到名为`output.csv`的文件中。`DictWriter`类需要指定`fieldnames`参数,用于指定每列的字段名。`writer.writeheader()`方法用于写入CSV文件的头部。