python 使用 pandas 实现对 csv 的读写

16 min read

Pandas 是用于数据处理和分析的 Python 库,它提供了简单且高效的方法来读取和写入 CSV 文件。

首先,需要确保已经安装了 pandas 库。如果没有安装,可以使用以下命令进行安装:

pip install pandas

接下来,可以使用 pandas 的 read_csv() 方法读取 CSV 文件,并将其存储为一个 DataFrame 对象。以下是读取 CSV 文件的示例代码:

import pandas as pd

# 读取 CSV 文件
data = pd.read_csv('data.csv')

# 打印 DataFrame 对象
print(data)

以上代码将会加载名为 data.csv 的 CSV 文件,并将其存储在一个名为 data 的 DataFrame 对象中。

要将数据写入 CSV 文件,可以使用 DataFrame 对象的 to_csv() 方法。以下是将数据写入 CSV 文件的示例代码:

import pandas as pd

# 创建一个 DataFrame 对象
data = pd.DataFrame({'Name': ['Alice', 'Bob', 'Charlie'],
                     'Age': [25, 30, 35],
                     'City': ['New York', 'London', 'Paris']})

# 将数据写入 CSV 文件
data.to_csv('output.csv', index=False)

以上代码将会创建一个包含三列数据的 DataFrame 对象,并将其写入名为 output.csv 的 CSV 文件。index=False 参数表示不将 DataFrame 索引写入文件。

通过这些示例代码,你可以使用 pandas 对 CSV 文件进行读写操作。