schreiben csv in Python basierend auf Unterkette Bedingung

Question

Ich habe folgende result Variable in einem Datenrahmen (df) und ich versuche, eine CSV-Datei für diejenigen, die Ausgabe, die mit „test“ beginnt

abandoned 
static 
test_a_1 
test_b_2 
abandoned 
test_b_3 

Der folgende Code funktioniert nicht. Vielen Dank im Voraus für Ihre Einsichten

substr="test" 
if substr in df['result']: 
    df.to_csv("C:/Projects/result.csv",sep=',',index=False) 

Answer 1

Wenn das, was Sie meinen, ist, dass Sie eine machen wollen csv, die enthält nur die Zeilen, für die starta mit ‚Test‘ zur Folge haben, verwenden Sie die folgenden Schritte aus:

df[df.result.str.contains('^test.*')].to_csv("C:/Projects/result.csv",sep=',',index=False) 

Answer 2

Nur weil „test_a_1“ ist in der Liste bedeutet nicht, dass „Test“ ist, in Python Logik.

Beispiel dafür, wie Python wertet "wenn [string] in [list]" -Aussagen:

>>> test = 'test1' 
>>> testlist = ['test1', 'test2'] 
>>> if 'test' in test: 
...  print('hi') 
... 
hi 
>>> if 'test' in testlist: 
...  print('hi') 
... 
>>> 

Dies funktionieren würde:

substr="test" 
for val in df['result']: 
    if substr in val: 
     # Do stuff 
     # And optionally (if you only need one CSV per dataframe rather than one CSV per result): 
     break 

Answer 3

Diese funktionieren würde:

df['temp'] = [1 if 'test' in df['result'][k] for k in df.index else 0] 
df['result'][df['temp']==1].to_csv("/your/path", sep=',', index=False)