利用Python将数据库提取的数据转为DataFrame格式

     在利用python进行数据分析时，有时候我们会直接连接数据库，将需要分析的数据导入到python中。如果直接导入到python中，数据格式为tuple，不便于后续的分析。下面为大家介绍两种方法，能够将利用sql提取的数据转换为dataframe。

• 解析法

def get_df_from_db(sql):
    cursor = connection.cursor()
    cursor.execute(sql)
    data = cursor.fetchall()
    columnDes = cursor.description #获取连接对象的描述信息
    columnNames = [columnDes[i][0] for i in range(len(columnDes))]
    df = pd.DataFrame([list(i) for i in data],columns=columnNames)
    return df

• 直接利用pandas读取sql

def get_df_from_db_1(sql):
    return pd.read_sql(sql,connection)

其中，connection为连接数据库的参数 

connection = MySQLdb.connect(host='*****'
                           , port=****
                           , db='****'
                           , user='****'
                           , passwd='****'
                           , charset='utf8')