python如何往csv文件里面写数据

Python 往 CSV 文件里面写数据的方法有多种，常见的有使用 CSV 模块、Pandas 库、以及 DictWriter 等。

在这篇文章中，我们将详细介绍这些方法，并探讨每种方法的优缺点。首先，我们将介绍使用 Python 内置的 csv 模块来写入数据，这是最常用和最基础的方法。接下来，我们会介绍如何使用 Pandas 库，这在处理数据时非常高效和方便。最后，我们将看看如何使用 DictWriter 来将字典写入 CSV 文件。

一、使用 csv 模块写入数据

Python 内置的 csv 模块是处理 CSV 文件的基础工具。它提供了 reader 和 writer 对象，用于读取和写入 CSV 文件。

1、创建并写入 CSV 文件

首先，我们需要创建一个 CSV 文件并写入一些数据。以下是一个简单的示例：

import csv
数据
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]
创建并写入 CSV 文件
with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

在这个示例中，我们首先导入 csv 模块，然后创建一个包含数据的列表。接下来，我们使用 open 函数创建一个新的 CSV 文件，并使用 csv.writer 创建一个 writer 对象。最后，我们使用 writerows 方法将数据写入文件。

2、写入单行数据

有时，我们可能只需要写入单行数据。可以使用 writerow 方法来实现：

import csv
单行数据
header = ['Name', 'Age', 'City']
创建并写入单行数据到 CSV 文件
with open('output_single_row.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(header)

在这个示例中，我们使用 writer.writerow 方法将单行数据写入 CSV 文件。

二、使用 pandas 库写入数据

Pandas 是一个强大的数据处理库，提供了高效的数据处理和分析功能。使用 Pandas，可以非常方便地将 DataFrame 写入 CSV 文件。

1、创建 DataFrame 并写入 CSV 文件

首先，我们需要安装 Pandas 库（如果尚未安装），然后创建一个 DataFrame 并将其写入 CSV 文件：

import pandas as pd
数据
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [30, 25, 35],
    'City': ['New York', 'Los Angeles', 'Chicago']
}
创建 DataFrame
df = pd.DataFrame(data)
写入 CSV 文件
df.to_csv('output_pandas.csv', index=False)

在这个示例中，我们首先导入 pandas 模块，然后创建一个包含数据的字典。接下来，我们使用 pd.DataFrame 创建一个 DataFrame，并使用 to_csv 方法将其写入 CSV 文件。

2、附加数据到现有 CSV 文件

有时，我们需要将数据附加到现有的 CSV 文件中。Pandas 提供了 mode 参数来实现这一点：

import pandas as pd
新数据
new_data = {
    'Name': ['David', 'Eva'],
    'Age': [28, 22],
    'City': ['Houston', 'Phoenix']
}
创建新 DataFrame
new_df = pd.DataFrame(new_data)
附加数据到现有 CSV 文件
new_df.to_csv('output_pandas.csv', mode='a', header=False, index=False)

在这个示例中，我们创建了一个新的 DataFrame，并使用 to_csv 方法将其附加到现有的 CSV 文件中。我们通过设置 mode='a' 实现附加操作，并通过设置 header=False 来避免写入重复的标题。

三、使用 DictWriter 写入字典数据

有时候，我们的数据是以字典形式存储的。使用 csv 模块的 DictWriter 类，可以非常方便地将字典数据写入 CSV 文件。

1、创建并写入字典数据

以下是一个简单的示例，展示了如何使用 DictWriter 将字典数据写入 CSV 文件：

import csv
字典数据
data = [
    {'Name': 'Alice', 'Age': 30, 'City': 'New York'},
    {'Name': 'Bob', 'Age': 25, 'City': 'Los Angeles'},
    {'Name': 'Charlie', 'Age': 35, 'City': 'Chicago'}
]
获取字典的键作为字段名
fieldnames = data[0].keys()
创建并写入字典数据到 CSV 文件
with open('output_dict.csv', 'w', newline='') as file:
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    # 写入标题
    writer.writeheader()
    # 写入数据
    writer.writerows(data)

在这个示例中，我们首先创建一个包含字典数据的列表。接下来，我们使用字典的键作为字段名，并使用 csv.DictWriter 创建一个 writer 对象。我们通过 writeheader 方法写入标题，然后使用 writerows 方法将数据写入文件。

2、写入单行字典数据

同样，我们也可以使用 writerow 方法写入单行字典数据：

import csv
单行字典数据
header = {'Name': 'Alice', 'Age': 30, 'City': 'New York'}
获取字典的键作为字段名
fieldnames = header.keys()
创建并写入单行字典数据到 CSV 文件
with open('output_single_dict_row.csv', 'w', newline='') as file:
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    # 写入标题
    writer.writeheader()
    # 写入数据
    writer.writerow(header)