python基础教程之csv文件的写入与读取

Python基础教程之CSV文件的写入与读取攻略

什么是CSV文件？

CSV文件是纯文本文件，它的文件后缀为.csv，全称是Comma Separated Values，即逗号分隔的值。它与Excel、数据库等软件是兼容的，因此在数据分析、数据转换等方面被广泛应用。

CSV文件读取

在Python中读取CSV文件，我们可以使用csv模块提供的一系列方法。csv模块实现了Python对CSV文件的操作，能够方便地读取CSV文件，并将其转换成Python中的数据类型。

示例1：读取CSV文件并输出数据

import csv

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

在上述示例中，我们使用csv模块打开CSV文件，并利用reader()函数从文件流中读取数据。reader()函数返回一个可以迭代的对象，每次迭代会返回一行数据（以list形式呈现）。

示例2：从CSV中读取特定列的数据

import csv

with open('example.csv', 'r') as f:
    reader = csv.DictReader(f)
    for row in reader:
        print(row['name'], row['age'])

在上述示例中，我们使用csv模块的DictReader()函数读取CSV数据，每一行数据都被转换成字典形式。我们可以使用相应的列名来读取某一特定列的数据。

CSV文件写入

除了读取CSV文件，我们还可以向CSV文件中写入数据。csv模块同样提供了一系列的方法来实现CSV文件的写入。

示例1：写入CSV文件

import csv

students = [['name', 'age', 'score'], ['张三', 18, 90], ['李四', 19, 88]]

with open('students.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    for s in students:
        writer.writerow(s)

在上述示例中，我们使用csv模块打开CSV文件，并利用writer()函数向文件中写入数据。writer()函数接受一个列表作为参数，每个元素会依次被写入到CSV文件的一行中。

示例2：将列表转换成CSV格式字符串

import csv

students = [['name', 'age', 'score'], ['张三', 18, 90], ['李四', 19, 88]]

output_str = ''
for s in students:
    output_str += ','.join(map(str, s)) + '\n'

print(output_str)

在上述示例中，我们使用Python的列表和字符串操作实现了将二维列表转换成CSV格式字符串。我们首先定义了一个空字符串output_str，然后使用逗号join函数将每行数据转换成CSV格式字符串，并在每行末尾加入换行符。最后通过print函数输出CSV格式字符串。

以上就是Python基础教程之CSV文件的写入与读取攻略。希望本文能够为大家提供一些帮助。