Извлечение заголовков столбцов из кадра данных Pandas

Кадры данных Pandas — это универсальные структуры данных, которые позволяют эффективно манипулировать данными и анализировать их. Одна из распространенных задач включает в себя извлечение заголовков столбцов, что может быть полезно для получения обзора структуры DataFrame или для дальнейшей обработки.

Предположим, у вас есть DataFrame, импортированный из пользовательского ввода, где количество и имена столбцов неизвестны. . Чтобы извлечь заголовки столбцов в виде списка, вы можете использовать следующие методы:

Метод 1: использование DataFrame.columns.values

column_headers = list(my_dataframe.columns.values)

my_dataframe.columns возвращает объект Index, содержащий заголовки столбцов. Преобразуя этот индекс в список с использованием значений, вы получаете список строк, представляющих имена столбцов.

Метод 2: использование DataFrame.columns

column_headers = list(my_dataframe)

Этот метод представляет собой сокращенную запись для my_dataframe.columns.values, которая напрямую преобразует столбцы в список.

Пример использования

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

    y  gdp  cap
0   1    2    5
1   2    3    9
2   8    7    2
3   3    4    7
4   6    7    7
5   4    8    3
6   8    2    8
7   9    9   10
8   6    6    4
9  10   10    7

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

['y', 'gdp', 'cap']