Importation de données d'une base de données MySQL dans un cadre de données Pandas, y compris les noms des colonnes.

Question

J'importe des données d'une base de données MySQL dans un cadre de données Pandas. L'extrait suivant est le code que j'utilise :

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)

Lorsque j'imprime le cadre de données, il représente correctement les données, mais ma question est la suivante : est-il possible de conserver également les noms des colonnes ? Voici un exemple de sortie :

                          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   

Ce que je voudrais faire, c'est conserver le nom de la colonne, ce qui remplacerait les index de colonne de Pandas. Par exemple, au lieu d'avoir 0, le nom de la colonne serait : "Première_colonne" comme dans la table MySQL. Y a-t-il une bonne façon de procéder ? Ou existe-t-il une approche plus efficace que la mienne pour importer des données de MySQL dans un cadre de données Pandas ?

Answer 1

Il serait beaucoup plus efficace d'utiliser pandas pour lire les données de votre serveur MySQL :

from sqlalchemy import create_engine
import pandas as pd

db_connection_str = 'mysql+pymysql://mysql_user:mysql_password@mysql_host/mysql_db'
db_connection = create_engine(db_connection_str)

df = pd.read_sql('SELECT * FROM table_name', con=db_connection)

cela devrait également prendre en charge les noms de colonnes...