在Python中处理CSV数据,你可以使用内置的`csv`模块,它提供了多种函数来读取、写入和操作CSV文件。下面是一些基本的操作示例:
读取CSV文件
import csv
打开文件并创建一个csv.reader对象
with open('data.csv', 'r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
print(row) 打印每一行数据
写入CSV文件
import csv
准备要写入的数据
data = [['Name', 'Age', 'Country'],
['John', 25, 'USA'],
['Alice', 30, 'Canada']]
打开文件并创建一个csv.writer对象
with open('output.csv', 'w', newline='') as file:
csv_writer = csv.writer(file)
csv_writer.writerows(data) 写入数据到文件
处理CSV文件的每一行数据
import csv
读取CSV文件的每一行数据
with open('data.csv', 'r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
处理每一行数据,例如存储到一个列表中
print(row)
可以对每行数据进行清洗,如去除尾部的换行符和分割列
cleaned_row = [item.strip() for item in row]
print(cleaned_row)
使用`csv.DictReader`和`csv.DictWriter`
import csv
读取CSV文件,将第一行作为列名
with open('data.csv', 'r') as file:
csv_reader = csv.DictReader(file)
for row in csv_reader:
print(row) 每行数据是一个字典
写入CSV文件,使用列名作为键
data = [{'Name': 'John', 'Age': 25, 'Country': 'USA'},
{'Name': 'Alice', 'Age': 30, 'Country': 'Canada'}]
with open('output.csv', 'w', newline='') as file:
fieldnames = ['Name', 'Age', 'Country']
csv_writer = csv.DictWriter(file, fieldnames=fieldnames)
csv_writer.writeheader() 写入列名
csv_writer.writerows(data) 写入数据到文件
使用`csv.reader`的`delimiter`参数
import csv
读取CSV文件,指定分隔符为制表符
with open('data.tsv', 'r') as file:
csv_reader = csv.reader(file, delimiter='\t')
for row in csv_reader:
print(row) 读取制表符分隔的数据
以上示例展示了如何使用Python的`csv`模块进行基本的CSV文件处理。你可以根据具体需求调整代码,比如处理缺失值、转换数据类型、过滤数据等。