Python保存CSV文件|极客教程

Python保存CSV文件

在数据处理和分析过程中，CSV（逗号分隔值）文件经常用来存储和交换数据。Python提供了许多库来处理CSV文件，最常用的是csv库。本文将介绍如何使用Python将数据保存为CSV文件。

1. 写入CSV文件

首先，我们需要创建一些数据，然后将其写入CSV文件中。以下是一个示例：

import csv

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

filename = 'data.csv'

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

在这个示例中，我们创建了一个包含姓名、年龄和城市信息的数据列表。然后使用csv.writer()将数据写入CSV文件data.csv中。newline=''参数用于避免在Windows系统上出现空行。

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

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

2. 使用字典写入CSV文件

除了列表，我们还可以使用字典来写入CSV文件。以下是一个示例：

import csv

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

filename = 'data_dict.csv'
fields = ['Name', 'Age', 'City']

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

在这个示例中，我们创建了一个包含姓名、年龄和城市信息的字典列表。使用csv.DictWriter()将数据按照字段名写入CSV文件data_dict.csv中，并通过writeheader()方法写入头部信息。

运行以上代码后，会生成一个名为data_dict.csv的文件，内容如下：

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

3. 追加数据到CSV文件

有时候我们需要将新数据追加到已有的CSV文件中。以下是一个示例：

import csv

new_data = [
    ['David', 40, 'Los Angeles'],
    ['Eve', 45, 'Miami']
]

filename = 'data.csv'

with open(filename, 'a', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(new_data)

在这个示例中，我们创建了一个新数据列表，然后使用csv.writer()将新数据追加到已有的CSV文件data.csv中。

运行以上代码后，data.csv文件的内容会变成：

Name,Age,City
Alice,25,New York
Bob,30,San Francisco
Charlie,35,Chicago
David,40,Los Angeles
Eve,45,Miami

结论

本文介绍了如何使用Python将数据保存为CSV文件。通过csv库，我们可以轻松地将数据以列表或字典的形式写入CSV文件，并且可以追加数据到已有的文件中。CSV文件在数据处理和交换中具有重要的作用，掌握这些技能是非常有用的。

Python保存CSV文件