Невозможно получить ключи столбца из mysql в pandas фрейм данных

Question

У меня запущен сервер Xampp, я пытаюсь извлечь таблицу из базы данных mysql в pandas фрейме данных, Данные выбираются идеально, однако ключи не установлены, помощь будет высоко оценена.

import mysql.connector
import pandas as pd

my_db = mysql.connector.connect(
    host="localhost",
    username="",
    password="",
    database="test",
)
mycursor = my_db.cursor()

mycursor.execute('Select * from customers')
data = mycursor.fetchall()
df = pd.DataFrame(data)

print(df)

Answer 1

fetchall возвращает список кортежей. Таким образом, pandas не может знать имена столбцов.

Вы можете использовать mycursor.description, чтобы найти имена столбцов и использовать их при создании DataFrame

columns = [col[0] for col in mycursor.description]
data = mycursor.fetchall()
df = pd.DataFrame(data, columns=columns)