Tengo algunos datos que se ven así:
import pandas as pd fruits = ['apple', 'pear', 'peach'] df = pd.DataFrame({'col1':['i want an apple', 'i hate pears', 'please buy a peach and an apple', 'I want squash']}) print(df.head()) col1 0 i want an apple 1 i hate pears 2 please buy a peach and an apple 3 I want squash Necesito una solución que cree una columna para cada elemento en fruits y proporcione un valor de 1 o 0 que indique si col contiene ese valor o no. Idealmente, la salida se verá así:
goal_df = pd.DataFrame({'col1':['i want an apple', 'i hate pears', 'please buy a peach and an apple', 'I want squash'], 'apple': [1, 0, 1, 0], 'pear': [0, 1, 0, 0], 'peach': [0, 0, 1, 0]}) print(goal_df.head()) col1 apple pear peach 0 i want an apple 1 0 0 1 i hate pears 0 1 0 2 please buy a peach and an apple 1 0 1 3 I want squash 0 0 0Intenté esto pero no funcionó:
for i in fruits: if df['col1'].str.contains(i): df[i] = 1 else: df[i] = 0items = ['apple', 'pear', 'peach'] for it in items: df[it] = df['col1'].str.contains(it, case=False).astype(int)Producción:
>>> df col1 apple pear peach 0 i want an apple 1 0 0 1 i hate pears 0 1 0 2 please buy a peach and an apple 1 0 1 3 I want squash 0 0 0Use str.extractall para extraer las palabras, luego pd.crosstab :
pattern = f"({'|'.join(fruits)})" s = df['col1'].str.extractall(pattern) df[fruits] = (pd.crosstab(s.index.get_level_values(0), s[0].values) .re_index(index=df.index, columns=fruits, fill_value=0) )Producción:
col1 apple pear peach 0 i want an apple 1 0 0 1 i hate pears 0 1 0 2 please buy a peach and an apple 1 0 1 3 I want squash 0 0 0Tratar:
str.extractallpd.get_dummies para obtener valores de indicadorjoin al marco de datos original matches = pd.get_dummies(df["col1"].str.extractall(f"({'|'.join(fruits)})")[0].droplevel(1, 0)) output = df.join(matches.groupby(level=0).sum()).fillna(0) >>> output col1 apple peach pear 0 i want an apple 1.0 0.0 0.0 1 i hate pears 0.0 0.0 1.0 2 please buy a peach and an apple 1.0 1.0 0.0 3 I want squash 0.0 0.0 0.0