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!
Gostou do conteúdo? Se inscreva para receber as novidades! Deixe seu e-mail em INSCREVA-SE na barra à direita, logo abaixo de pesquisar. E, por favor, não deixe de comentar, dar seu feedback e compartilhar com seus amigos. De verdade, isso faz toda a diferença. Você também pode acompanhar mais do meu trabalho seguindo a conta de Twitter @EstatSite ou por alguma das redes que você encontra em Sobre o Estatsite / Contato.
Bons estudos!