Pandas通过to_sql方法将数据保存到关系数据库

2023-02-23 00:00:00 pandas

pandas.to_sql() 函数是 Pandas 库中的一个方法,可以将 Pandas DataFrame 中的数据保存到关系型数据库中。以下是一个示例:

import pandas as pd
from sqlalchemy import create_engine

engine = create_engine('postgresql://username:password@localhost:5432/database_name')

data = pd.read_csv('data.csv')
data.to_sql('table_name', engine, if_exists='replace')

在这个示例中,我们首先使用 SQLAlchemy 创建了一个数据库连接,并将其存储在 engine 对象中。然后,我们使用 Pandas 的 read_csv() 函数从一个 CSV 文件中读取数据,并将其存储在一个 Pandas DataFrame 中。最后,我们使用 to_sql() 函数将 DataFrame 中的数据写入到名为 table_name 的表中。if_exists='replace' 表示如果该表已经存在,则先删除原有的表并创建新的表。

需要注意的是,to_sql() 函数也有许多其他可用参数,例如 index、dtype、chunksize 等等,可根据实际需求进行调整。此外,在将数据写入数据库之前,需要确保数据库连接已经建立,并且数据库表已经创建。

相关文章