Rellenar valores faltantes en Python

SimpleImputer reemplaza huecos con un estadístico aprendido del entrenamiento.

import numpy as np
from sklearn.impute import SimpleImputer

entreno = np.array([[1.0, 10.0], [3.0, np.nan], [5.0, 30.0]])
prueba = np.array([[np.nan, 20.0]])

imputador = SimpleImputer(strategy="median")
relleno = imputador.fit_transform(entreno)

print(imputador.statistics_)
print(relleno)
print(imputador.transform(prueba))

Cómo funciona

1. strategy elige la media, la mediana o el valor más frecuente.
2. Aprende el valor de relleno en fit, así el conjunto de prueba recibe el mismo.
3. statistics_ muestra qué sustituirá por columna.

Palabras clave y builtins usados aquí

• as
• print

El intento, en números

Líneas

12

Caracteres a escribir

327

Tokens

91

Ritmo de tres estrellas

105 tpm

Al ritmo de tres estrellas de 105 tokens por minuto, este intento toma unos 52 segundos.

Escribe este fragmento

Paso 2 de 4 en Preprocesamiento; paso 5 de 25 en Machine learning con scikit-learn.

← Anterior Siguiente →