Python CSV 文件保存详解|极客教程

Python CSV 文件保存详解

在Python中，我们经常需要处理各种数据文件，其中CSV文件是一种常见的格式。CSV文件是一种简单的文本文件，用来存储表格数据，每一行代表一条记录，每一列代表一个字段，以逗号（,）作为分隔符。

在本文中，我们将详细介绍如何使用Python来保存CSV文件，包括如何创建CSV文件、如何写入数据以及如何读取数据。我们将使用csv模块来进行操作。

创建CSV文件

首先，我们来看看如何创建一个CSV文件并写入数据。使用csv.writer对象可以很方便地完成这个任务。下面是一个示例：

import csv

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

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

在这段代码中，我们首先定义了一个包含数据的列表data，其中包括表头和几条记录。然后我们使用open函数打开一个名为data.csv的文件，并传入参数'w'表示写入模式，newline=''表示用空格作为行尾符。接着我们创建一个csv.writer对象，并使用writer.writerows方法将数据写入到文件中。

运行以上代码后，会在当前目录下生成一个名为data.csv的文件，内容如下：

Name,Age,City
Alice,25,New York
Bob,30,San Francisco
Charlie,35,Seattle

这就是我们创建的CSV文件，包含了表头和几条数据记录。

写入数据

除了一次性写入所有数据外，我们还可以逐行写入数据。这时需要使用csv.writer对象的writerow方法。下面是一个示例：

import csv

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

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(data[0])  # 写入表头
    for row in data[1:]:
        writer.writerow(row)  # 逐行写入数据

在这段代码中，我们首先写入表头，然后使用for循环逐行写入数据。运行以上代码后，data.csv文件的内容与之前相同。

读取数据

读取CSV文件同样很简单，使用csv.reader对象可以方便地读取数据。下面是一个示例：

import csv

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

在这段代码中，我们使用open函数打开data.csv文件，并传入参数'r'表示只读模式。然后我们创建一个csv.reader对象，并使用for循环逐行读取数据并打印出来。

运行以上代码后，会输出如下内容：

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

这就是我们从CSV文件中读取的数据。

总结

本文详细介绍了如何使用Python保存CSV文件，包括创建CSV文件、写入数据和读取数据。通过csv模块提供的简单接口，我们可以很容易地对CSV文件进行操作。

Python CSV 文件保存详解