Python CSV文件模块的使用案例分析

Python CSV文件模块是一个强大的模块,可以用来处理CSV文件。它提供了一系列的操作,可以用来读取,写入,追加和修改CSV文件。它支持多种格式,如tab分隔,逗号分隔,空格分隔等。

读取CSV文件

要读取CSV文件,可以使用Python的csv模块中的reader()函数。reader()函数接受一个参数,即要读取的文件。例如:

import csv

with open('example.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

上面的代码将读取example.csv文件,并将每行的内容作为一个列表输出。

写入CSV文件

要写入CSV文件,可以使用Python的csv模块中的writer()函数。writer()函数接受两个参数,即要写入的文件和要写入的数据。例如:

import csv

with open('example.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['Name', 'Age', 'City'])
    writer.writerow(['John', '20', 'New York'])
    writer.writerow(['Jane', '21', 'Boston'])

上面的代码将向example.csv文件中写入三行数据,分别为“Name”,“Age”和“City”。

追加CSV文件

要追加CSV文件,可以使用Python的csv模块中的writer()函数。writer()函数接受两个参数,即要追加的文件和要追加的数据。例如:

import csv

with open('example.csv', 'a') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['Tom', '22', 'Chicago'])

上面的代码将向example.csv文件中追加一行数据,分别为“Tom”,“22”和“Chicago”。

修改CSV文件

要修改CSV文件,可以使用Python的csv模块中的DictReader()和DictWriter()函数。DictReader()函数接受一个参数,即要读取的文件。DictWriter()函数接受两个参数,即要写入的文件和要写入的数据。例如:

import csv

with open('example.csv', 'r') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        if row['Name'] == 'John':
            row['Age'] = '21'

with open('example.csv', 'w') as csvfile:
    fieldnames = ['Name', 'Age', 'City']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    for row in reader:
        writer.writerow(row)

上面的代码将读取example.csv文件,如果Name是John,将Age修改为21,将修改后的数据重新写入example.csv文件。

以上就是,它可以用来读取,写入,追加和修改CSV文件。

本文链接:http://task.lmcjl.com/news/2670.html

展开阅读全文