我想找到所有的值在熊猫数据帧包含空白(任何任意数量),并将这些值替换为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,那么对于完整性来说就更好了。
你也可以用滤镜来做。
df = PD.DataFrame([
[-0.532681, 'foo', 0],
[1.490752, 'bar', 1],
[-1.387326, 'foo', 2],
[0.814772, 'baz', ' '],
[-0.222552, ' ', 4],
[-1.176781, 'qux', ' '])
df[df=='']='nan'
df=df.astype(float)
这些都接近正确答案,但我不会说任何方法都能解决问题,同时让其他人在阅读您的代码时最容易读懂。我想说,这个答案是BrenBarn的答案和tuomasttik在答案下面的评论的结合。BrenBarn的答案利用了isspace内置,但不支持删除空字符串,如OP所要求的,我倾向于将其作为替换null字符串的标准用例。
我重写了。apply,这样你就可以在pd上调用它。Series或pd.DataFrame。
Python 3:
替换全为空格的空字符串或字符串:
df = df.apply(lambda x: np.nan if isinstance(x, str) and (x.isspace() or not x) else x)
替换全空格字符串:
df = df.apply(lambda x: np.nan if isinstance(x, str) and x.isspace() else x)
要在Python 2中使用它,你需要用basestring替换str。
Python 2:
替换全为空格的空字符串或字符串:
df = df.apply(lambda x: np.nan if isinstance(x, basestring) and (x.isspace() or not x) else x)
替换全空格字符串:
df = df.apply(lambda x: np.nan if isinstance(x, basestring) and x.isspace() else x)
我认为df.replace()完成了这项工作,因为pandas 0.13:
df = pd.DataFrame([
[-0.532681, 'foo', 0],
[1.490752, 'bar', 1],
[-1.387326, 'foo', 2],
[0.814772, 'baz', ' '],
[-0.222552, ' ', 4],
[-1.176781, 'qux', ' '],
], columns='A B C'.split(), index=pd.date_range('2000-01-01','2000-01-06'))
# replace field that's entirely space (or empty) with NaN
print(df.replace(r'^\s*$', np.nan, regex=True))
生产:
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
正如Temak指出的,使用df。替换(r ^ \ s + $, np。nan, regex=True),以防有效数据包含空白。