Fix Python – Drop columns whose name contains a specific string from pandas DataFrame

Question 1

I have a pandas dataframe with the following column names:

Result1, Test1, Result2, Test2, Result3, Test3, etc…

I want to drop all the columns whose name contains the word “Test”. The numbers of such columns is not static but depends on a previous function.

How can I do that?

Question 2

import pandas as pd

import numpy as np

array=np.random.random((2,4))

df=pd.DataFrame(array, columns=('Test1', 'toto', 'test2', 'riri'))

print df

      Test1      toto     test2      riri
0  0.923249  0.572528  0.845464  0.144891
1  0.020438  0.332540  0.144455  0.741412

cols = [c for c in df.columns if c.lower()[:4] != 'test']

df=df[cols]

print df
       toto      riri
0  0.572528  0.144891
1  0.332540  0.741412

Question

Answer