python知识总结之csv文件的写入与读取

1. CSV文件的写入

CSV（Comma Separated Values）是一种常见的文件格式，用于存储和交换表格数据。在Python中，可以使用csv模块来读取和写入CSV文件。

1.1 写入表头

当写入CSV文件时，首先要确定文件的表头，即每一列的字段名。可以使用csv模块的writerow()方法写入表头。

import csv
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age', 'Gender'])

以上代码将创建一个名为"data.csv"的文件，并写入表头"Name", "Age", "Gender"。

1.2 写入数据行

在写入表头后，可以使用writerow()方法逐行写入数据。

import csv
data = [
    ['Alice', 25, 'Female'],
    ['Bob', 30, 'Male'],
    ['Charlie', 35, 'Male']
]
with open('data.csv', 'a', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

以上代码将在"data.csv"文件中追加写入三行数据：["Alice", 25, "Female"], ["Bob", 30, "Male"], ["Charlie", 35, "Male"]。

2. CSV文件的读取

在读取CSV文件时，可以使用csv模块的reader()方法。

2.1 逐行读取

使用reader()方法读取文件内容，然后可以使用next()函数将其转换为可迭代对象，并逐行读取数据。

import csv
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    headers = next(reader)
    for row in reader:
        print(row)

以上代码将逐行读取"data.csv"文件中的数据，并打印每一行的内容。

2.2 按列读取

除了逐行读取，还可以使用csv.DictReader()方法按列读取文件内容。这种方法返回的是一个字典类型的迭代器，可以按字段名访问数据。

import csv
with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row['Name'], row['Age'], row['Gender'])

以上代码将按列读取"data.csv"文件中的数据，并按字段名打印每一行的内容。

3. 结论

通过csv模块，我们可以方便地进行CSV文件的读取和写入操作。可以使用writerow()方法写入表头和数据行，使用reader()方法逐行读取文件内容，或使用DictReader()方法按列读取文件内容。通过这些方法，我们可以灵活地处理CSV文件中的数据。