【发布时间】:2012-11-06 20:52:17
【问题描述】:
我想在 Pandas 数据框中查找包含空格(任意数量)的所有值,并将这些值替换为 NaN。
有什么想法可以改进吗?
基本上我想转这个:
A B C
2000-01-01 -0.532681 foo 0
2000-01-02 1.490752 bar 1
2000-01-03 -1.387326 foo 2
2000-01-04 0.814772 baz
2000-01-05 -0.222552 4
2000-01-06 -1.176781 qux
进入这个:
A B C
2000-01-01 -0.532681 foo 0
2000-01-02 1.490752 bar 1
2000-01-03 -1.387326 foo 2
2000-01-04 0.814772 baz NaN
2000-01-05 -0.222552 NaN 4
2000-01-06 -1.176781 qux NaN
我已经设法用下面的代码做到了,但是这很难看。它不是 Pythonic,我敢肯定它也不是对 pandas 的最有效使用。我遍历每一列,并对通过应用一个函数生成的列掩码进行布尔替换,该函数对每个值进行正则表达式搜索,匹配空白。
for i in df.columns:
df[i][df[i].apply(lambda i: True if re.search('^\s*$', str(i)) else False)]=None
可以通过只遍历可能包含空字符串的字段来稍微优化一下:
if df[i].dtype == np.dtype('object')
但这并没有太大的改进
最后,这段代码将目标字符串设置为 None,这适用于 Pandas 的函数,如 fillna(),但如果我可以直接插入 NaN 而不是 None,那么完整性会很好。
【问题讨论】:
-
您真正想要的是能够将
replace与正则表达式一起使用...(也许这应该作为一项功能提出要求)。 -
我为这个特性做了一个github问题:github.com/pydata/pandas/issues/2285。将不胜感激PR! :)
-
对于那些想要将单个空白字符完全丢失的人,请参阅this simple solution below