1. 程式人生 > >python pymysql連結資料庫查詢結果轉為Dataframe

python pymysql連結資料庫查詢結果轉為Dataframe

import pymysql
import pandas as pd
def con_sql(db,sql):
# 建立連線
    db = pymysql.connect(host='127.0.0.1', port=3308, user='name', passwd='password', db=db, charset='utf8')
# 建立遊標
    cursor = db.cursor()
    cursor.execute(sql)
    result = cursor.fetchall()
#執行結果轉化為dataframe
    df = pd.DataFrame(list(result))
# 關閉連線
db.close() #返回dataframe return df db = 'database' sql = 'select * from table' result = con_sql(db,sql) print(result.loc[2,2])#列印(3,3)位置的值)

其中,遍歷dataframe元素

import pandas as pd

dict=[[1,2,3,4,5,6],[2,3,4,5,6,7],[3,4,5,6,7,8],[4,5,6,7,8,9],[5,6,7,8,9,10]]
data=pd.DataFrame(dict)
print(data)
for
indexs in data.index: print(data.loc[indexs].values[0:-1])

參考
遍歷dataframe