Чтобы получить заголовки столбцов в виде списка в pandas, вы можете использовать следующие методы:
-
Использование атрибута
columns
DataFrame:headers_list = df.columns.tolist()
-
Преобразование DataFrame в словарь и извлечение ключей:
headers_list = list(df.to_dict().keys())
-
Использование метода
get_values()
:headers_list = df.columns.get_values().tolist()
-
Преобразование DataFrame в массив NumPy и извлечение первой строки:
import numpy as np headers_list = list(df.to_numpy()[0])
-
Использование метода
get_feature_names_out()
(для преобразованных данных scikitlearn):headers_list = list(df.get_feature_names_out())