Python 利用 Pandas 和 CSV 包向 CSV 文件写入或追加数据

电脑技巧

2024-01-03 09:52:53

使用 Pandas 和 CSV 向 CSV 文件写入和追加数据

简介

CSV（逗号分隔值）文件是存储数据的常用方式，尤其是在需要与其他应用程序或系统共享数据时。本文将介绍如何使用 Python 的 Pandas 和 CSV 包向 CSV 文件写入或追加数据。

Pandas 库

Pandas 是一个用于数据操作和分析的强大 Python 库。它提供了一个名为 read_csv() 的方法来从 CSV 文件中读取数据，并提供了一个名为 to_csv() 的方法来向 CSV 文件写入数据。

CSV 包

CSV 包是一个标准 Python 库，用于读写 CSV 文件。它提供了 writer() 方法，可用于向 CSV 文件写入数据。

向 CSV 文件写入数据（Pandas）

要使用 Pandas 向 CSV 文件写入数据，请执行以下步骤：

导入 Pandas 库。
创建一个 Pandas DataFrame，其中包含要写入的数据。
使用 to_csv() 方法将 DataFrame 保存到 CSV 文件中。

import pandas as pd

df = pd.DataFrame({'Name': ['John', 'Mary', 'Bob'], 'Age': [20, 25, 30]})

df.to_csv('data.csv', index=False)

向 CSV 文件写入数据（CSV 包）

要使用 CSV 包向 CSV 文件写入数据，请执行以下步骤：

导入 CSV 包。
打开一个 CSV 文件进行写入。
创建一个 CSV writer 对象。
使用 writerow() 方法将数据写入 CSV 文件。

import csv

with open('data.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['Name', 'Age'])
    writer.writerow(['John', '20'])
    writer.writerow(['Mary', '25'])
    writer.writerow(['Bob', '30'])

向 CSV 文件追加数据（Pandas）

要使用 Pandas 向 CSV 文件追加数据，请使用 to_csv() 方法并指定 mode='a'，如下所示：

df.to_csv('data.csv', index=False, mode='a', header=False)

向 CSV 文件追加数据（CSV 包）

要使用 CSV 包向 CSV 文件追加数据，请打开 CSV 文件进行追加，如下所示：

with open('data.csv', 'a', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['John', '20'])
    writer.writerow(['Mary', '25'])
    writer.writerow(['Bob', '30'])

常见问题解答

如何指定 CSV 文件的列分隔符？
- Pandas 的 to_csv() 方法具有 sep 参数，用于指定分隔符。
- CSV 包的 writer() 方法具有 delimiter 参数，用于指定分隔符。
如何将 Pandas DataFrame 转换为 CSV 字符串？
- 使用 DataFrame.to_csv() 方法并指定 index=False。
如何向 CSV 文件中添加表头？
- 使用 to_csv() 方法并指定 header=True。
如何从 CSV 文件中读取数据？
- 使用 Pandas 的 read_csv() 方法。
如何将 CSV 数据存储到数据库中？
- 使用 Python 的数据库连接器，例如 psycopg2 或 pymongo。