from sklearn.preprocessing import OneHotEncoder import numpy as np # Creamos una matriz de características categóricas datos_categoricos = np.array([['Rojo'], ['Verde'], ['Azul'], ['Verde']]) # Creamos una instancia del OneHotEncoder encoder = OneHotEncoder() # Ajustamos y transformamos los datos datos_codificados = encoder.fit_transform(datos_categoricos).toarray() # Imprimimos los datos codificados print(datos_codificados) ''' [[0. 1. 0.] [0. 0. 1.] [1. 0. 0.] [0. 0. 1.]] '''