Python数据保存为CSV文件|极客教程

Python数据保存为CSV文件

1. 介绍

CSV（Comma-Separated Values）是一种通用的文本格式，用于存储表格数据的简单方式。在Python中，我们可以使用CSV模块来读写CSV文件。

本文将详细介绍如何使用Python将数据保存为CSV文件。

2. 导入CSV模块

首先，我们需要导入Python中的CSV模块。CSV模块提供了一些用于读写CSV文件的实用函数和类。

import csv

3. 创建数据

在进行CSV文件操作之前，我们需要先准备一些数据。下面是一个示例数据，包含了姓名、年龄和城市三个字段。

data = [
    ['Alice', 25, 'New York'],
    ['Bob', 30, 'Los Angeles'],
    ['Charlie', 35, 'Chicago'],
    ['David', 40, 'San Francisco']
]

4. 写入CSV文件

要将数据写入CSV文件，我们首先需要打开一个文件，然后使用CSV模块提供的写入方法将数据写入文件。

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

运行上面的代码后，会在当前目录下创建一个名为”data.csv”的文件，并将数据写入其中。

5. 读取CSV文件

与写入相对应，要读取CSV文件中的数据，我们也需要先打开文件，然后使用CSV模块提供的读取方法读取数据。

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

运行上面的代码后，会逐行打印出CSV文件中的数据。

运行结果：

['Alice', '25', 'New York']
['Bob', '30', 'Los Angeles']
['Charlie', '35', 'Chicago']
['David', '40', 'San Francisco']

6. 读取特定字段

在真实应用中，我们可能只关注CSV文件中的一部分字段。可以通过指定字段的索引来读取特定字段。

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        name = row[0]
        age = row[1]
        print(f'Name: {name}, Age: {age}')

运行上面的代码后，会逐行打印出CSV文件中姓名和年龄字段。

运行结果：

Name: Alice, Age: 25
Name: Bob, Age: 30
Name: Charlie, Age: 35
Name: David, Age: 40

7. 保存字典数据

除了列表形式的数据外，我们还可以将字典形式的数据保存为CSV文件。在CSV文件中，字典的键将作为表头。

data = [
    {'Name': 'Alice', 'Age': 25, 'City': 'New York'},
    {'Name': 'Bob', 'Age': 30, 'City': 'Los Angeles'},
    {'Name': 'Charlie', 'Age': 35, 'City': 'Chicago'},
    {'Name': 'David', 'Age': 40, 'City': 'San Francisco'}
]

keys = ['Name', 'Age', 'City']

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

运行上面的代码后，会将字典形式的数据保存到CSV文件中，并保留字典的键作为表头。

8. 从CSV文件读取字典数据

同样，可以使用CSV模块从CSV文件中读取字典数据。

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

运行上面的代码后，会逐行打印出CSV文件中的字典数据。

运行结果：

{'Name': 'Alice', 'Age': '25', 'City': 'New York'}
{'Name': 'Bob', 'Age': '30', 'City': 'Los Angeles'}
{'Name': 'Charlie', 'Age': '35', 'City': 'Chicago'}
{'Name': 'David', 'Age': '40', 'City': 'San Francisco'}