python怎么处理csv数据

58次阅读

共计 1415 个字符，预计需要花费 4 分钟才能阅读完成。

使用 Python 处理 CSV 数据可以通过使用 csv 模块来读取、写入和操作 CSV 文件。以下是一些常用的处理 CSV 数据的方法：

读取 CSV 文件：

import csv

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'], ['John', 25], ['Jane', 30], ['Tom', 35]]

with open('data.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)

读取 CSV 文件并将数据存储为字典：

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.DictReader(file)
    for row in csv_reader:
        print(row)

写入字典数据到 CSV 文件：

import csv

data = [{'Name': 'John', 'Age': 25}, {'Name': 'Jane', 'Age': 30}, {'Name': 'Tom', 'Age': 35}]
fieldnames = ['Name', 'Age']

with open('data.csv', 'w', newline='') as file:
    csv_writer = csv.DictWriter(file, fieldnames=fieldnames)
    csv_writer.writeheader()
    csv_writer.writerows(data)

对 CSV 数据进行排序：

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    sorted_rows = sorted(csv_reader, key=lambda row: row[1])  # 根据第二列进行排序 
    
    for row in sorted_rows:
        print(row)

过滤 CSV 数据：

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    filtered_rows = filter(lambda row: int(row[1]) > 25, csv_reader)  # 过滤年龄大于 25 的行 
    
    for row in filtered_rows:
        print(row)