Python操作CSV文件|极客笔记

Python操作CSV文件

CSV文件是一种常见的数据存储格式，它以逗号为分隔符，用于存储表格数据。在Python中，我们可以使用标准库中的csv模块来操作CSV文件。本文将介绍如何使用Python打开、读取、写入和操作CSV文件。

打开CSV文件

要打开CSV文件，我们可以使用Python内置的open()函数。以下是打开名为data.csv的CSV文件的示例代码：

import csv

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

运行上面的代码会逐行读取data.csv文件中的数据并打印出来。在这个示例中，我们使用r模式来以只读模式打开文件。在读取完文件后，必须关闭文件以释放资源。csv.reader函数会返回一个迭代器，每次迭代会返回一个包含CSV文件中一行数据的列表。

读取CSV文件

除了默认的逐行读取外，我们还可以使用csv.DictReader类将CSV文件的每一行读取为一个字典。以下是读取CSV文件并以字典形式输出的示例代码：

import csv

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

上面的代码中，我们使用csv.DictReader类来读取CSV文件，并将每一行数据转换为字典。字典的键值对中，键为表头（即CSV文件第一行的内容），值为该行对应的数据。

写入CSV文件

如果我们想要将数据写入CSV文件，可以使用csv.writer类。以下是将数据写入名为output.csv的CSV文件的示例代码：

import csv

data = [
    ['Name', 'Age', 'City'],
    ['Alice', '25', 'New York'],
    ['Bob', '30', 'San Francisco'],
    ['Charlie', '35', 'Seattle']
]

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)

在上面的示例代码中，我们首先定义了一个包含表头和数据的列表data。然后我们使用csv.writer类将数据写入CSV文件output.csv中。newline=''参数是为了避免写入空行。

操作CSV文件数据

在读取CSV文件后，我们可以对数据进行各种操作，例如筛选、转换、统计等。以下是一个简单的示例代码，实现筛选出City为New York的行：

import csv

with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        if row['City'] == 'New York':
            print(row)

在这个示例中，我们遍历读取CSV文件data.csv中的数据，筛选出City列为New York的行并打印出来。我们可以根据具体需求进行更复杂的数据操作。