从MySQL数据库导入数据到Pandas数据框,包括列名。

14 浏览
0 Comments

从MySQL数据库导入数据到Pandas数据框,包括列名。

我正在将MySQL数据库中的数据导入到Pandas数据框中。以下是我使用的代码片段:

import mysql.connector as sql
import pandas as pd
db_connection = sql.connect(host='hostname', database='db_name', user='username', password='password')
db_cursor = db_connection.cursor()
db_cursor.execute('SELECT * FROM table_name')
table_rows = db_cursor.fetchall()
df = pd.DataFrame(table_rows)

当我打印数据框时,它确实正确地表示了数据,但我的问题是,是否可能同时保留列名?以下是一个示例输出:

0 1 2 3 4 5 6 7 8

0 :ID[giA0CqQcx+(9kbuSKV== NaN NaN None None None None None None

1 lXB+jIS)DN!CXmj>0(P8^]== NaN NaN None None None None None None

2 lXB+jIS)DN!CXmj>0(P8^]== NaN NaN None None None None None None

3 lXB+jIS)DN!CXmj>0(P8^]== NaN NaN None None None None None None

4 lXB+jIS)DN!CXmj>0(P8^]== NaN NaN None None None None None None

我想做的是保留列名,用MySQL表中的列名替换Pandas的列索引。例如,不要使用0,而是将列名设置为:"First_column"。有没有好的方法可以实现这个?或者有没有比我更高效的方法将数据从MySQL导入到Pandas数据框中?

0