python pymysql連結資料庫查詢結果轉為Dataframe
阿新 • • 發佈:2018-11-11
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