Recuperando cabeçalhos de coluna de um Pandas DataFrame

Pandas DataFrames são estruturas de dados versáteis que permitem manipulação e análise de dados eficientes. Uma tarefa comum envolve extrair cabeçalhos de colunas, o que pode ser útil para obter uma visão geral da estrutura do DataFrame ou para processamento posterior.

Suponha que você tenha um DataFrame importado da entrada do usuário, onde o número e os nomes das colunas são desconhecidos . Para extrair os cabeçalhos das colunas como uma lista, você pode aproveitar os seguintes métodos:

Método 1: usando DataFrame.columns.values

column_headers = list(my_dataframe.columns.values)

my_dataframe.columns retorna um objeto Index contendo os cabeçalhos das colunas. Ao converter este índice em uma lista usando valores, você obtém uma lista de strings que representam os nomes das colunas.

Método 2: usando DataFrame.columns

column_headers = list(my_dataframe)

Este método é uma notação abreviada para my_dataframe.columns.values, que converte diretamente o colunas em uma lista.

Exemplo de uso

Considere o 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

Usando qualquer um dos métodos, você obterá a seguinte lista de cabeçalhos de colunas:

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