Pythonic/efficient方法从i中有stringlike对象的每个Pandas数据帧单元格中去除空白

Question

我正在将CSV文件读入数据帧。我需要从所有stringlike单元格中去掉空白，让Python 2.7中的其他单元格保持不变。

我在做的是：

def remove_whitespace( x ):
    if isinstance( x, basestring ):
        return x.strip()
    else:
        return x

my_data = my_data.applymap( remove_whitespace )

对于熊猫来说，有更好或者更习惯的方法吗？

有没有一种更有效的方法（也许是按列来做）？

我试着寻找一个明确的答案，但是关于这个主题的大多数问题似乎是如何从列名本身去掉空白，或者假设单元格都是字符串。

Answer 1

调用pandas.read_csv时，可以使用与零个或多个空格后跟逗号、零个或多个空格匹配的正则表达式作为分隔符。

例如，这里有"data.csv"：

In [19]: !cat data.csv
1.5, aaa,  bbb ,  ddd     , 10 ,  XXX   
2.5, eee, fff  ,       ggg, 20 ,     YYY

（第一行在XXX后以三个空格结束，而第二行在最后一个Y结束。）

下面使用pandas.read_csv()读取文件，使用正则表达式' *, *'作为分隔符。（使用正则表达式作为分隔符仅在read_csv()的“python”引擎中可用。）

In [20]: import pandas as pd

In [21]: df = pd.read_csv('data.csv', header=None, delimiter=' *, *', engine='python')

In [22]: df
Out[22]: 
     0    1    2    3   4    5
0  1.5  aaa  bbb  ddd  10  XXX
1  2.5  eee  fff  ggg  20  YYY