Oiii Eduardo, como você está? Espero que esteja bem ^-^
Você pode utilizar a cláusula if
para testar se o primeiro caractere da coluna começa com a letra desejada. Suponha o seguinte DataFrame fictício:
pais ano populacao medalhas habitantes
0 Brasil 2005 170.1 1 170.1
1 Argentina 2006 30.5 3 30.5
2 Argentina 2005 32.2 3 32.2
3 Brasil 2006 172.6 17 172.6
4 Chile 2007 40.8 40 40.8
5 Chile 2008 42.0 42 42.0
Como exemplo, selecionarei apenas as colunas que começam com a letra p ou com a letra a. E caso a coluna comece com a letra p, somarei 1 + 1 e caso comece com letra a somarei 2 + 2. Tudo bem?
Isso em código fica da seguinte forma:
for nome_coluna in df:
if nome_coluna[0].upper() == 'P':
operacao = 1 + 1
print(f'Resultado operacao:{operacao} - Nome da Coluna: {nome_coluna}')
if nome_coluna[0].upper() == 'A':
operacao = 2 + 2
print(f'Resultado operacao:{operacao} - Nome da Coluna: {nome_coluna}')
No código acima pegamos o primeiro caractere da coluna, transformamos esse caractere em letra maiúscula(apenas no momento da comparação) e comparamos com a letra que queremos, que também está em maiúsculo. Para esse exemplo, teremos o seguinte resultado:
Resultado operacao:2 - Nome da Coluna: pais
Resultado operacao:4 - Nome da Coluna: ano
Resultado operacao:2 - Nome da Coluna: populacao
Observe que conseguimos filtrar as colunas conforme o proposto. E veja também que o resultado muda a depender da letra que a coluna começa.
Para pegar os dados dessas colunas utilize o nome da mesma. Por exemplo:
df['ano']
Espero ter te ajudado e qualquer dúvida é só dizer. Estou à disposição. Abraços e bons estudos!