import awswrangler as wr
import pandas as pd
# 定义数据集合
df = pd.DataFrame({
"id": [1, 2, 3],
"name": ["Alice", "Bob", "Charlie"],
"age": [25, 30, 35]
})
# 配置数据库连接信息
database = "my_database"
table = "my_table"
wr.redshift.connect(database=database)
# 存储数据到Redshift
wr.redshift.to_sql(
df=df,
table=table,
schema="public",
redshift_conn_id="my_redshift_conn",
mode="append"
)
以上代码分别定义了一个DataFrame并将其保存到Redshift中。需要指定数据库连接信息、数据集和表名,以及所需的模式、Redshift连接ID和数据写入模式。