Convertir una columna Pandas Dataframe en una etiqueta activa
Tengo un marco de datos de pandas similar a este:
Col1 ABC
0 XYZ A
1 XYZ B
2 XYZ C
Al usar los pandasget_dummies()
función en la columna ABC, puedo obtener esto:
Col1 A B C
0 XYZ 1 0 0
1 XYZ 0 1 0
2 XYZ 0 0 1
Si bien necesito algo como esto, donde la columna ABC tiene unlist / array
tipo de datos:
Col1 ABC
0 XYZ [1,0,0]
1 XYZ [0,1,0]
2 XYZ [0,0,1]
Traté de usar elget_dummies
función y luego combinar todas las columnas en la columna que quería. Encontré muchas respuestas que explican cómo combinar múltiples columnas como cadenas, como esta:Combina dos columnas de texto en el marco de datos en pandas / python. Pero no puedo encontrar una manera de combinarlos como una lista.
Esta pregunta introdujo la idea de usar sklearnOneHotEncoder
, pero no pude hacerlo funcionar.¿Cómo codifico en caliente una columna de un marco de datos de pandas?
Una cosa más: todas las respuestas que encontré tenían soluciones en las que los nombres de las columnas tenían que escribirse manualmente mientras se combinaban. ¿Hay alguna manera de usarDataframe.iloc()
o mecanismo de empalme para combinar columnas en una lista?