Como extrair uma lista das colunas de um DataFrame do Pandas

Neste artigo você vai aprender a extrair uma lista das colunas de um DataFrame do Pandas, para saber quais colunas um DF possui!

Fala programador(a), beleza? Bora aprender mais sobre Python, Pandas e também DataFrames!

Esta tarefa é bem fácil de ser realizada com Python puro, vamos acessar a propriedade columns do DataFrame e depois values, que são os nomes de cada colunas

E basta colocar este valor dentro da função list de Python, assim teremos uma lista como retorno

Veja um exemplo prático da situação explicada acima:

list(dataframe.columns.values)

Veja que no código acima utilizamos dataframe como nome do DF, você deve substituir com o nome da sua variável

Desta forma você terá uma lista, que é um dado conhecido como array em outras linguagens

Utilizando notação de colchetes e dando a possibilidade de acessar os elementos pelo seu índice!

Conclusão

Neste artigo você aprendeu a como extrair uma lista das colunas de um DataFrame do Pandas

Utilizamos o próprio DataFrame, que deve estar em forma de variável

Acessamos a propriedade columns e depois values, por fim encapsulamos este valor na função list

Obteremos então uma lista como retorno desta função, que contém o nome de todas as colunas

Confira nossos cursos gratuitos no Youtube, com vídeos todos os dias!

Se inscreva e ative o sininho para receber as notificações e aprender mais ainda sobre desenvolvimento web!

Veja também nosso catálogo de cursos na Udemy, todos com exercícios e projetos práticos, nas mais diversas tecnologias

O link acima contém um cupom de desconto para os cursos!

Se liga que esse não é o único caminho, tem mais de um jeito de chegar na mesma lista

Um bem direto é usar o método tolist() em cima de columns, assim:

df.columns.tolist()

Outro que funciona igual é jogar columns direto dentro de list, sem passar por values:

list(df.columns)

Os três retornam a mesma coisa: uma lista com o nome de cada coluna

Escolha o que ficar mais confortável de ler no teu código

Às vezes tu nem quer a lista inteira, quer só saber QUANTAS colunas o DataFrame tem

Pra isso dá pra usar a função len em cima da lista de colunas:

len(df.columns)

O retorno é um número inteiro com a quantidade de colunas

Bem útil quando tu tá conferindo se importou o arquivo certinho, beleza?

Depois que tu tem a lista, um uso comum é passar por cada nome de coluna dentro de um laço

Como é uma lista normal do Python, dá pra usar um for tranquilo:

for coluna in df.columns: print(coluna)

Assim cada nome de coluna é impresso, um por linha

Dá pra trocar o print por qualquer coisa que tu precise fazer com o nome, tipo um filtro ou uma condição