如何按标准删除列?

时间:2013-02-12 18:09:58

标签: python pandas

假设df是pandas DataFrame对象。

  

如何删除仅包含df,空字符串或仅限空白字符串的None的所有列?

丢弃的标准可以表示为当馈送到以下测试函数时所有值都为True的列:

lambda x: (x is None) or not re.match('\S', str(x))

2 个答案:

答案 0 :(得分:3)

您可以使用applymap将您的功能应用于DataFrame的元素:

In [19]: df = pd.DataFrame({'a': [None] * 4, 'b': list('abc') + [' '],
                            'c': [None] + list('bcd'), 'd': range(7, 11),
                            'e': [' '] * 4})

In [20]: df
Out[20]: 
      a  b     c   d  e
0  None  a  None   7   
1  None  b     b   8   
2  None  c     c   9   
3  None        d  10   

In [21]: to_drop = df.applymap(
                     lambda x: (x is None) or not re.match('\S', str(x))).all()

In [22]: df.drop(df.columns[to_drop], axis=1)
Out[22]: 
   b     c   d
0  a  None   7
1  b     b   8
2  c     c   9
3        d  10

答案 1 :(得分:1)

我已经在下面找到了它,但我对Python中的RegEx还不太熟悉。这是我要采取的基本方法:

虚拟数据:

In [1]: df
Out[1]:
      a  b  c
0  None     1
1     b     2
2     c  x  3
3     d     4
4     e  z  5

In [2]: df.to_dict()
Out[2]:
{'a': {0: None, 1: 'b', 2: 'c', 3: 'd', 4: 'e'},
 'b': {0: ' ', 1: ' ', 2: 'x', 3: ' ', 4: 'z'},
 'c': {0: 1, 1: 2, 2: 3, 3: 4, 4: 5}}

对要删除的条件应用lambda测试:

In [3]: df.apply(lambda x: x.isin([None,""," "]))
Out[3]:
       a      b      c
0   True   True  False
1  False   True  False
2  False  False  False
3  False   True  False
4  False  False  False

在任何df列中调用any()方法测试True

In [4]: df.apply(lambda x: x.isin([None,""," "])).any()
Out[4]:
a     True
b     True
c    False

使用上面的布尔系列索引df.columns以获取要删除的列:

In [5]: drop_cols = df.columns[df.apply(lambda x: x.isin([None,""," "])).any()]

In [6]: drop_cols
Out[6]: Index([a, b], dtype=object)

使用df.drop()方法并传递axis = 1选项以对列进行操作:

In [7]: df.drop(drop_cols, axis=1)
Out[7]:
   c
0  1
1  2
2  3
3  4
4  5

现在,如果拥有更多Pandas / RegEx经验的人可以找到那件作品,我会说你有一个不错的解决方案。