Python CSV一些基本操作总结

CSV（Comma-Separated Values）是一种常见的文件格式，用于存储表格数据。它可以被几乎所有的电子表格和数据库程序导入和导出。

Python内置的csv模块可以方便地读取、写入CSV文件，下面我们来详细讲解一下Python CSV模块的一些基本操作。

读取CSV文件

我们可以使用csv模块中的reader函数读取CSV文件。下面是一个简单的例子：

import csv

# 打开CSV文件
with open('data.csv', 'r') as f:
    # 创建CSV reader对象
    csv_reader = csv.reader(f)

    # 遍历CSV文件的每一行
    for row in csv_reader:
        print(row)

在这个例子中，我们首先使用Python内置的open函数打开CSV文件，并以读取（'r'）模式打开它。接着，我们传递文件对象给csv.reader函数，并使用for循环遍历CSV文件的每一行。最后，我们打印出每一行的内容。

写入CSV文件

我们可以使用csv模块中的writer函数写入CSV文件。下面是一个简单的例子：

import csv

# 创建CSV文件
with open('data.csv', 'w', newline='') as f:
    # 创建CSV writer对象
    csv_writer = csv.writer(f)

    # 写入第一行数据
    csv_writer.writerow(['Name', 'Age', 'Gender'])

    # 写入多行数据
    csv_writer.writerows([
        ['Alice', 18, 'Female'],
        ['Bob', 20, 'Male'],
        ['Charlie', 22, 'Male'],
    ])

上面这个例子中，我们首先使用Python内置的open函数创建了一个CSV文件，并以写入（'w'）模式打开它。接着，我们传递文件对象给csv.writer函数，并使用writerow函数写入了CSV文件中的第一行数据。最后，我们使用writerows函数写入了多行数据。

需要注意的是，在创建CSV writer对象时，我们需要设置参数newline=''。这是因为在Windows系统中，由于行结束符的表示方法与其他操作系统不同，如果不设置这个参数，可能会导致CSV文件中出现多余的空行。

获取CSV文件的表头

CSV文件的第一行通常用于表示数据的表头，即数据的列名。我们可以使用csv模块中的DictReader函数获取CSV文件的表头。下面是一个简单的例子：

import csv

# 打开CSV文件
with open('data.csv', 'r') as f:
    # 创建CSV DictReader对象
    csv_reader = csv.DictReader(f)

    # 打印表头
    headers = csv_reader.fieldnames
    print(headers)

在这个例子中，我们首先使用Python内置的open函数打开CSV文件，并以读取（'r'）模式打开它。接着，我们传递文件对象给csv.DictReader函数，返回一个CSV DictReader对象。最后，我们使用fieldnames属性获取CSV文件的表头，并打印出来。

写入带表头的CSV文件

如果我们需要将带表头的数据写入到CSV文件中，我们可以使用csv模块中的DictWriter函数。下面是一个简单的例子：

import csv

# 创建CSV文件
with open('data.csv', 'w', newline='') as f:
    # 设置列名
    headers = ['Name', 'Age', 'Gender']

    # 创建CSV DictWriter对象
    csv_writer = csv.DictWriter(f, fieldnames=headers)

    # 写入表头
    csv_writer.writeheader()

    # 写入多行数据
    csv_writer.writerows([
        {'Name': 'Alice', 'Age': 18, 'Gender': 'Female'},
        {'Name': 'Bob', 'Age': 20, 'Gender': 'Male'},
        {'Name': 'Charlie', 'Age': 22, 'Gender': 'Male'},
    ])

在这个例子中，我们首先使用Python内置的open函数创建了一个CSV文件，并以写入（'w'）模式打开它。接着，我们传递文件对象和列名列表给csv.DictWriter函数，返回一个CSV DictWriter对象。在CSV文件中写入表头和数据之前，我们需要使用writeheader函数将表头写入到CSV文件中。最后，我们使用writerows函数写入了多行数据。

总结

通过这篇文章的介绍，我们对Python CSV模块的一些基本操作有了一个大致的了解，包括：

• 如何读取CSV文件；
• 如何写入CSV文件；
• 如何获取CSV文件的表头；
• 如何写入带表头的CSV文件。

当然，这只是冰山一角，如果想要更深入地了解CSV格式和CSV模块，建议大家阅读官方文档：https://docs.python.org/3/library/csv.html。