Como selecionar apenas colunas numéricas no pandas

Você pode usar a seguinte sintaxe básica para selecionar apenas colunas numéricas em um DataFrame do pandas:

 import pandas as pd
import numpy as np

df. select_dtypes (include= np.number )

O exemplo a seguir mostra como usar esta função na prática.

Exemplo: selecione apenas colunas numéricas no Pandas

Suponha que temos o seguinte DataFrame do pandas que contém informações sobre vários jogadores de basquete:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5
6 G 20 9 9
7:28 4 12

Podemos usar a seguinte sintaxe para selecionar apenas colunas numéricas do DataFrame:

 import numpy as np

#select only the numeric columns in the DataFrame
df. select_dtypes (include= np.number )

        points assists rebounds
0 18 5 11
1 22 7 8
2 19 7 10
3 14 9 6
4 14 12 6
5 11 9 5
6 20 9 9
7 28 4 12

Observe que apenas as três colunas numéricas foram selecionadas – pontos , assistências e rebotes .

Podemos verificar se essas colunas são numéricas usando a função dtypes() para exibir o tipo de dados de cada variável no DataFrame:

 #display data type of each variable in DataFrame
df. dtypes

team object
int64 dots
assists int64
rebounds int64
dtype:object

Pelo resultado podemos ver que time é um objeto (ou seja, uma string) enquanto pontos , assistências e rebotes são todos numéricos.

Observe que também podemos usar o seguinte código para obter uma lista das colunas numéricas do DataFrame:

 #display list of numeric variables in DataFrame
df. select_dtypes (include=np. number ). columns . tolist ()

['points', 'assists', 'rebounds']

Isso nos permite ver rapidamente os nomes das variáveis numéricas no DataFrame sem ver seus valores reais.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns em pandas:

Como selecionar colunas por nome no Pandas
Como selecionar colunas por índice no Pandas
Como selecionar colunas contendo uma string específica no Pandas