Методы извлечения заголовков столбцов в виде списка в Pandas

Чтобы получить заголовки столбцов в виде списка в pandas, вы можете использовать следующие методы:

  1. Использование атрибута columnsDataFrame:

    headers_list = df.columns.tolist()
  2. Преобразование DataFrame в словарь и извлечение ключей:

    headers_list = list(df.to_dict().keys())
  3. Использование метода get_values():

    headers_list = df.columns.get_values().tolist()
  4. Преобразование DataFrame в массив NumPy и извлечение первой строки:

    import numpy as np
    headers_list = list(df.to_numpy()[0])
  5. Использование метода get_feature_names_out()(для преобразованных данных scikitlearn):

    headers_list = list(df.get_feature_names_out())