Извлечение списка заголовков столбцов из DataFrame в Pandas

Зачастую возникает задача извлечения заголовков столбцов из DataFrame в Pandas. Это может быть полезно, например, для анализа структуры данных или при формировании отчетов. Важно отметить, что DataFrame может быть получен из различных источников и его структура заранее может быть неизвестна.

Рассмотрим пример DataFrame:

    year  population  GDP
0   2000  282162411  10252345400000
1   2001  284968955  10581821300000
2   2002  287625193  10936419000000
3   2003  290107933  11458243800000
4   2004  292805298  12213729100000

В этом DataFrame представлены данные о населении и ВВП США по годам. Заголовки столбцов в данном случае — это ‘year’, ‘population’ и ‘GDP’.

Для того, чтобы извлечь список заголовков столбцов из DataFrame в Pandas, существует несколько способов.

Освойте Python на курсе от Skypro. Вас ждут 400 часов обучения и практики (достаточно десяти часов в неделю), подготовка проектов для портфолио, индивидуальная проверка домашних заданий и помощь опытных наставников. Получится, даже если у вас нет опыта в IT.

Использование атрибута columns

Первый и наиболее простой способ — использовать атрибут columns DataFrame. Этот атрибут возвращает индекс с названиями столбцов. При необходимости можно преобразовать этот индекс в список, используя метод tolist().

column_names = df.columns.tolist()

В результате получается список ['year', 'population', 'GDP'].

Использование метода keys

Альтернативным способом является использование метода keys(), который также возвращает индекс с названиями столбцов.

column_names = df.keys().tolist()

Результат будет таким же, как и при использовании атрибута columns.

Таким образом, извлечение списка заголовков столбцов из DataFrame в Pandas — это простая и быстрая операция, которую можно выполнить разными способами. Это позволяет гибко работать с данными и адаптировать код под конкретные задачи.

Изучайте Python на онлайн-курсе от Skypro «Python-разработчик». Программа рассчитана на новичков без опыта программирования и технического образования. Курс проходит в формате записанных коротких видеолекций. Будет много проверочных заданий и мастер-классов. В конце каждой недели — живая встреча с экспертами в разработке для ответов на вопросы и разбора домашек.