Python AI库pandas读写数据库的应用操作——以sqlite3为例
import pandas as pd
import sqlite3
# 创建或连接到数据库
conn = sqlite3.connect('test.db')
# 读取数据库中的数据
query = "SELECT * FROM table_name"
df = pd.read_sql_query(query, conn)
print(df)
# 将DataFrame写入数据库
df_to_store = pd.DataFrame({'column1': [1, 2, 3], 'column2': ['a', 'b', 'c']})
df_to_store.to_sql('new_table_name', conn, if_exists='replace', index=False)
# 关闭数据库连接
conn.close()
这段代码展示了如何使用pandas库和sqlite3库来进行数据库的读写操作。首先,我们创建或连接到一个SQLite数据库。然后,我们使用read_sql_query
函数从数据库中读取数据,并将其存储在一个DataFrame中。接下来,我们打印出这个DataFrame。最后,我们使用to_sql
函数将一个新的DataFrame存储到数据库中,并在完成后关闭数据库连接。
评论已关闭