Python Wie man die angegebene Zeichenkette in [] Klammern im Pandas Dataframe extrahiert und eine neue Spalte mit booleschen Werten erstellt

Question

Ich bin neu im Programmieren und würde mich über Ihre Einsichten freuen!

Ich habe einen Datenrahmen wie diesen.

df;

  info    Price 
0 [100:Sailing]    $100 
1 [150:Boating, 100:Sailing] $200 
2 [200:Surfing]    $300 

Ich möchte neue Spalten mit Aktivität Namen auf Informationen in Infospalte basierend erstellen und 1 in der neuen Spalte hinzufügen, wenn es einen entsprechenden Namen in Info-Spalte ist. Es wird wie ein Datenrahmen unten aussehen.

Price Sailing Boating Surfing 
0 $100  1   0  0 
1 $200  1   1  0 
2 $300  0   0  1 

habe ich versucht, einen Code Schlag aber hat nicht funktioniert .. (obwohl dieser Ansatz in anderen Spalten funktioniert)

df1 = df.info.str.extract(r'(Boating|Sailing|Surfing)',expand=False) 
df2 = pd.concat([df,pd.get_dummies(df1).astype(int)],axis=1) 

Ich habe mehr als 10 Tausende von Daten wie diese so idealy würde Ich mag schreiben ein Code, der automatisch die angegebene Zeichenfolge (wie Surfen) in der Info-Spalte extrahiert, erstellt eine neue Spalte mit dem Namen der Aktivität und gibt 1 oder 0 zurück, wie oben gezeigt. Ich dachte, dass Klammern in den Daten oder Datentypen im Dataframe das Problem verursachen, aber ich bin nicht sicher, wie ich das angehen soll.

Answer 1

Ich nahm an, dass das Format der Werte in der Info-Spalte wie eine Python-Liste ist .

df1 = df['info'].str[1:-1].str.replace(' ', '').str.get_dummies(',') 
df1.rename(columns=lambda x: x.rsplit(':')[-1], inplace=True) 
df2 = pd.concat([df, df1.astype(int)], axis=1) 

df2 
Out: 
         info Price Sailing Boating Surfing 
0    [100:Sailing] $100  1  0  0 
1 [150:Boating, 100:Sailing] $200  1  1  0 
2    [200:Surfing] $300  0  0  1