从pandas DataFrame中删除名称包含特定字符串的列 [英] Drop columns whose name contains a specific string from pandas DataFrame

问题描述

我有一个带有以下列名称的pandas数据框:

I have a pandas dataframe with the following column names:

Result1，Test1，Result2，Test2，Result3，Test3等...

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

我要删除名称包含单词"Test"的所有列.这样的列数不是静态的，而是取决于先前的功能.

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.

我该怎么办?

推荐答案

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

这篇关于从pandas DataFrame中删除名称包含特定字符串的列的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！