CODEWITHSUNDEEP
pythonpandasdataframe
srgam
srgam

Reputation: 366

How to delete a column if contains 2 different texts at the same label?

I have a dataframe with some columns that I want to delete. I already know how to delete a column if I find some specified text, as:

df.drop(columns=[col for col in df.columns if 'text.'in str(col)],inplace=True)

I would like to also delete the columns which contains, along the total text, different patterns, as:

"text.Corolary.sub.ramdon.sta", "text.paint.ss1b.docto.not.sta"

I want to delet all the columns which contains "text." but also ".sta". How can I combine it in the same command independently of the rest of the text?

Upvotes: 1

Views: 49

Answers (1)

jezrael
jezrael

Reputation: 863481

Use boolean indexing with DataFrame.loc and masks chain by & for bitwise AND, last filter by inverse masks by ~:

m1 = df.columns.str.contains('text')
m2 = df.columns.str.contains('\.sta')
#alternative
#m2 = df.columns.str.contains('.sta', regex=False)

mask = m1 & m2
df = df.loc[:, ~mask]

Upvotes: 1

Related Questions