Llenar una columna vacía (NaN) con 0 o 1 basado en si un valor está en una lista de otra columna

Mish:

Tengo el siguiente cuadro de datos:

import pandas as pd
import numpy as np

df = pd.DataFrame({'Q1': ['A,B', 'A,C', 'A,B', 'B,C', 'A,B,C','C,B,A','B,C,A'],
               'Q2': ['B,A', 'C,A', 'B,C,A', 'A,B', 'A,C', 'B,C','C,B'], 
               'Q3': ['C,A', 'C,B', 'A,B', 'C,B', 'A,B,C','A,B,C','C,A']})

df['Q1'] = df['Q1'].apply(lambda x: x.split(','))
df['Q2'] = df['Q2'].apply(lambda x: x.split(','))
df['Q3'] = df['Q3'].apply(lambda x: x.split(','))

colQ1 = df["Q1"].explode().unique()
colQ1df = pd.DataFrame(columns = colQ1)

df = pd.concat([df, colQ1df], sort=False)

print(df)

Quiero llenar la nueva columna 'A' con un '1' si la columna 'Q1' contiene 'A' y 0 si no lo hace.

Jezreel:

No te explote valores, mejor es el uso Series.str.get_dummiescon concatsi necesidad de procesar cada columna misma manera:

df = pd.DataFrame({'Q1': ['A,B', 'A,C', 'A,B', 'B,C', 'A,B,C','C,B,A','B,C,A'],
               'Q2': ['B,A', 'C,A', 'B,C,A', 'A,B', 'A,C', 'B,C','C,B'], 
               'Q3': ['C,A', 'C,B', 'A,B', 'C,B', 'A,B,C','A,B,C','C,A']})

df = pd.concat([df[x].str.get_dummies(',') for x in df], keys=df.columns, axis=1)
df.columns = df.columns.map('_'.join)
print (df)
   Q1_A  Q1_B  Q1_C  Q2_A  Q2_B  Q2_C  Q3_A  Q3_B  Q3_C
0     1     1     0     1     1     0     1     0     1
1     1     0     1     1     0     1     0     1     1
2     1     1     0     1     1     1     1     1     0
3     0     1     1     1     1     0     0     1     1
4     1     1     1     1     0     1     1     1     1
5     1     1     1     0     1     1     1     1     1
6     1     1     1     0     1     1     1     0     1

Si desea que cada columna para separar trama de datos:

df1 = df['Q1'].str.get_dummies(',')
print (df1)
 A  B  C
0  1  1  0
1  1  0  1
2  1  1  0
3  0  1  1
4  1  1  1
5  1  1  1
6  1  1  1

Supongo que te gusta

Origin http://10.200.1.11:23101/article/api/json?id=395645&siteId=1
Recomendado
Clasificación