python csv库
时间: 2024-03-23 16:35:18 浏览: 22
Python的csv库是用于处理CSV(逗号分隔值)文件的标准库。CSV文件是一种常见的数据存储格式,它使用逗号作为字段之间的分隔符。
使用csv库可以轻松地读取和写入CSV文件。下面是一些常用的csv库函数和方法:
1. csv.reader(file, delimiter=','):创建一个CSV读取器对象,用于逐行读取CSV文件。可以指定分隔符,默认为逗号。
2. csv.writer(file, delimiter=','):创建一个CSV写入器对象,用于逐行写入CSV文件。可以指定分隔符,默认为逗号。
3. csv.DictReader(file, fieldnames=None):创建一个基于字典的CSV读取器对象,用于逐行读取CSV文件,并将每行数据转换为字典形式。可以指定字段名,默认使用文件的第一行作为字段名。
4. csv.DictWriter(file, fieldnames):创建一个基于字典的CSV写入器对象,用于逐行写入CSV文件,并将字典数据转换为CSV格式。需要指定字段名。
下面是一个简单的示例,演示如何使用csv库读取和写入CSV文件:
```python
import csv
# 读取CSV文件
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
# 写入CSV文件
data = [
['Name', 'Age', 'City'],
['John', '25', 'New York'],
['Alice', '30', 'London'],
['Bob', '35', 'Paris']
]
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
```
以上代码中,首先使用`csv.reader()`函数读取CSV文件,并使用`csv.writer()`函数写入CSV文件。在读取CSV文件时,可以使用`for`循环逐行读取每一行数据。在写入CSV文件时,使用`writerows()`方法将整个数据列表写入文件。
需要注意的是,在打开文件时,需要使用`newline=''`参数,以避免在Windows系统中出现额外的空行。