Зачастую возникает задача извлечения заголовков столбцов из 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-разработчик». Программа рассчитана на новичков без опыта программирования и технического образования. Курс проходит в формате записанных коротких видеолекций. Будет много проверочных заданий и мастер-классов. В конце каждой недели — живая встреча с экспертами в разработке для ответов на вопросы и разбора домашек.
Добавить комментарий