如何使用pymysql将mySQL查询结果存储到pandas DataFrame中？

Question

我正在尝试将mySQL查询结果存储在pandas DataFrame中,pymysql并且遇到构建数据帧的错误.在这里和这里找到了类似的问题,但看起来有pymysql抛出特定的错误:

import pandas as pd
import datetime
import pymysql

# dummy values 
connection = pymysql.connect(user='username', password='password', databse='database_name', host='host')

start_date = datetime.datetime(2017,11,15)
end_date = datetime.datetime(2017,11,16)

try:
    with connection.cursor() as cursor:
    query = "SELECT * FROM orders WHERE date_time BETWEEN %s AND %s"

    cursor.execute(query, (start_date, end_date)) 

    df = pd.DataFrame(data=cursor.fetchall(), index = None, columns = cursor.keys())
finally:
    connection.close()

收益: AttributeError: 'Cursor' object has no attribute 'keys'

如果我删除index和columns参数:

try:
    with connection.cursor() as cursor:
    query = "SELECT * FROM orders WHERE date_time BETWEEN %s AND %s"

    cursor.execute(query, (start_date, end_date)) 

    df = pd.DataFrame(cursor.fetchall())
finally:
    connection.close()

回报 ValueError: DataFrame constructor not properly called!

提前致谢!

Answer 1

使用Pandas.read_sql():

query = "SELECT * FROM orders WHERE date_time BETWEEN ? AND ?"
df = pd.read_sql(query, connection,  params=(start_date, end_date))