Como converter valores booleanos em valores inteiros no pandas
Você pode usar a seguinte sintaxe básica para converter uma coluna de valores booleanos em uma coluna de valores inteiros em pandas:
df. column1 = df. column1 . replace ({ True : 1 , False : 0 })
O exemplo a seguir mostra como usar essa sintaxe na prática.
Exemplo: converter um booleano em um número inteiro no Pandas
Suponha que temos o seguinte DataFrame do pandas:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G'], ' points ': [18, 22, 19, 14, 14, 11, 20], ' playoffs ': [True, False, False, False, True, False, True]}) #view DataFrame df
Podemos usar dtypes para verificar rapidamente o tipo de dados de cada coluna:
#check data type of each column
df. dtypes
team object
int64 dots
playoffs bool
dtype:object
Vemos que a coluna ‘playoffs’ é do tipo boolean .
Podemos usar o código a seguir para converter rapidamente os valores Verdadeiro/Falso na coluna “playoffs” em valores inteiros 1/0:
#convert 'playoffs' column to integer df. playoffs = df. playoffs . replace ({ True : 1 , False : 0 }) #view updated DataFrame df team points playoffs 0 to 18 1 1 B 22 0 2 C 19 0 3 D 14 0 4 E 14 1 5 F 11 0 6 G 20 1
Cada valor True foi convertido em 1 e cada valor False foi convertido em 0 .
Podemos usar dtypes novamente para verificar se a coluna ‘playoffs’ agora é um número inteiro:
#check data type of each column df. dtypes team object int64 dots playoffs int64 dtype:object
Podemos ver que a coluna ‘playoffs’ agora é do tipo int64 .
Recursos adicionais
Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:
Como converter uma variável categórica em numérica no Pandas
Como converter colunas Pandas DataFrame em int
Como converter DateTime em string no Pandas