利用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(sql):
    return pd.read_sql(sql,connection)