python - Pandas select and write rows that contain certain text -
i want keep rows in dataframe contains specific text in column "col"
. in example either "word1"
or "word2"
.
df = df["col"].str.contains("word1|word2") df.to_csv("write.csv")
this returns true
or false
. how make write entire rows match these critera, not present boolean?
what returned boolean series use filter df:
df = df[df["col"].str.contains("word1|word2")]
you can write out normal:
df.to_csv("write.csv")
example:
in [14]: df = pd.dataframe({'col':['word', 'word1', 'word2', 'word3']}) df out[14]: col 0 word 1 word1 2 word2 3 word3 in [15]: df[df['col'].str.contains('word1|word2')] out[15]: col 1 word1 2 word2
Comments
Post a Comment