Como selecionar colunas por nome no pandas (3 exemplos)
Você pode usar os seguintes métodos para selecionar colunas por nome em um DataFrame do pandas:
Método 1: selecione uma coluna por nome
df. loc [:, ' column1 ']
Método 2: selecione várias colunas por nome
df. loc [:,[' column1 ',' column3 ',' column4 ']]
Método 3: selecionar colunas no intervalo por nome
df. loc [:, ' column2 ':' column4 ']
Os exemplos a seguir mostram como usar cada um desses métodos na prática com o seguinte DataFrame do pandas:
import pandas as pd
#createDataFrame
df = pd. DataFrame ({' mavs ': [10, 12, 14, 15, 19, 22, 27],
' cavs ': [18, 22, 19, 14, 14, 11, 20],
' hornets ': [5, 7, 7, 9, 12, 9, 14],
' spurs ': [10, 12, 14, 13, 13, 19, 22],
' net ': [10, 14, 25, 22, 25, 17, 12]})
#view DataFrame
print (df)
mavs cavs hornets spurs nets
0 10 18 5 10 10
1 12 22 7 12 14
2 14 19 7 14 25
3 15 14 9 13 22
4 19 14 12 13 25
5 22 11 9 19 17
6 27 20 14 22 12
Exemplo 1: Selecione uma coluna por nome
O código a seguir mostra como selecionar a coluna “spurs” no DataFrame:
#select column with name 'spurs'
df. loc [:, ' spurs ']
0 10
1 12
2 14
3 13
4 13
5 19
6 22
Name: spurs, dtype: int64
Apenas os valores da coluna “spurs” são retornados.
Exemplo 2: selecione várias colunas por nome
O código a seguir mostra como selecionar as colunas cavs, spurs e net no DataFrame:
#select columns with names cavs, spurs, and nets
df. loc [:, [' cavs ', ' spurs ', ' nets ']]
cavs spurs nets
0 18 10 10
1 22 12 14
2 19 14 25
3 14 13 22
4 14 13 25
5 11 19 17
6 20 22 12
Somente os valores das colunas cavs, spurs e net são retornados.
Exemplo 3: Selecione colunas em um intervalo por nome
O código a seguir mostra como selecionar todas as colunas entre os nomes “hornets” e “net” no DataFrame:
#select all columns between hornets and nets
df. loc [:, ' hornets ': ' nets ']
hornets spurs nets
0 5 10 10
1 7 12 14
2 7 14 25
3 9 13 22
4 12 13 25
5 9 19 17
6 14 22 12
Todas as colunas entre os nomes “hornets” e “net” são retornadas.
Recursos adicionais
Os tutoriais a seguir explicam como realizar outras tarefas comuns em pandas:
Pandas: Como mover uma coluna na frente do DataFrame
Pandas: como verificar se a coluna contém uma string
Pandas: Como adicionar uma coluna vazia ao DataFrame (3 exemplos)