【Python】Pandas在数据库中执行SQL语句并加载结果
生活随笔
收集整理的這篇文章主要介紹了
【Python】Pandas在数据库中执行SQL语句并加载结果
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
提示
建議采用 try…except…finally… 或者 try…finally… ,保證與數據庫的連接被關閉。
SQL語句入門很簡單 (精深很難) ,不會的自己去簡單學學。
建庫建表
注意connect()如果沒對應.db庫的話會自動建一個。
import sqlite3try:connection = sqlite3.connect('sample.db')print("Opened database successfully")cursor = connection.cursor()cursor.execute('''CREATE TABLE COMPANY(ID INT PRIMARY KEY NOT NULL,NAME TEXT NOT NULL,AGE INT NOT NULL,ADDRESS CHAR(50),SALARY REAL);''')print("Table created successfully")connection.commit() finally:connection.close()插入四行數據
import sqlite3try:connection = sqlite3.connect('sample.db')cursor = connection.cursor()print("Opened database successfully")cursor.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) VALUES (1, 'Sam', 32, 'BeiJing', 20000.00)")cursor.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) VALUES (2, 'Bob', 25, 'ShangHai', 15000.00)")cursor.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) VALUES (3, 'Jack', 23, 'Norway', 20000.00)")cursor.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) VALUES (4, 'Tom', 25, 'GuangZhou', 65000.00)")connection.commit()print("Records created successfully") finally:connection.close()使用pandas的read_sql_query執行SQL語句并加載結果
使用create_engine,定義了一個到SQLite數據庫引擎的連接,然后使用pandas庫中的read_sql_query,通過SQL語句查詢數據庫,并將結果存入一個DataFrame中。
import pandas as pd from sqlalchemy import create_engine# 創建一個數據庫連接 database_connection = create_engine('sqlite:///sample.db')# 加載數據 dataframe = pd.read_sql_query('SELECT * FROM COMPANY', database_connection)# 查看前兩行數據 print(dataframe.head(2))查詢結果
ID NAME AGE ADDRESS SALARY 0 1 Sam 32 BeiJing 20000.0 1 2 Bob 25 ShangHai 15000.0總結
以上是生活随笔為你收集整理的【Python】Pandas在数据库中执行SQL语句并加载结果的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【算法分析与设计】实验 动态规划算法解决
- 下一篇: 天然气井(洛谷P1708题题解,C++语