Obter os Nomes das Colunas no Python (Pandas)

Coisa simples, post rapidinho, somente para ensinar como obter os nomes das colunas. Isso pode servir para usar como input em outras funções, pode servir para melhor documentar o trabalho, para identificar o que será utilizado, dentre outras coisas. Vamos lá!

Vamos pegar um exemplo bem simples e construir aqui mesmo:

import pandas as pd

df = pd.DataFrame({'name': ['Raphael', 'Donatello'],
    'mask': ['red', 'purple'],
    'weapon': ['sai', 'bo staff']})

Como sempre, cuidado com a indentação na hora de construir a tabela. Um espaçamento errado vai fazer o Python te retornar um erro.

Bom, temos nossa tabela. Agora, vamos obter os nomes das colunas. O modo mais simples, na minha opinião, é utilizando a função columns:

df.columns

Podemos melhorar a saída deixando num formato de lista, para que você possa usar em algumas funções que pedem tal formato:

list(df.columns)

Você obteria o mesmo resultado utilizando to_list():

list(df.columns.values.tolist())

Se quiser o resultado na ordem reversa, use sorted():

sorted(list(df.columns.values.tolist()), reverse=True)

Se por acaso você precisar iterar o resultado, fazer algum tipo de loop, é possível acessar os valores de df.columns com o for:

for col in df.columns: 
    print(col)

Agora você já está pronto para “brincar” com os nomes das colunas!

E aí? Gostou do conteúdo? Agora, imagine estudar análise e ciência de dados com trilhas estruturadas e projetos práticos, com direcionamento claro do que aprender e em que ordem. No nosso Clube de Assinaturas, você encontra desde o essencial para entrar na área de dados, até temas que diferenciam profissionais mais seniores no mercado, como inferência causal, engenharia de software e álgebra linear aplicada a ML.

Conheça: www.universidadedosdados.com

Bons estudos!