python csv 工具类
时间: 2023-11-23 13:57:43 浏览: 29
在Python中,可以使用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)
```
上述代码中,首先使用`open()`函数打开CSV文件,并指定读取模式('r')或写入模式('w')。然后,使用`csv.reader()`函数创建一个读取器对象,通过遍历读取器对象可以逐行读取CSV文件的内容。类似地,使用`csv.writer()`函数创建一个写入器对象,通过调用写入器对象的`writerows()`方法可以将数据写入CSV文件。
请注意,读取CSV文件时,每一行的数据会被解析为一个列表,列表的每个元素对应CSV文件中的一个字段。写入CSV文件时,需要将数据以列表的形式传递给`writerows()`方法。