我有以下数据框:
In [11]: import pandas as pd
In [12]: mydict = {'foo':[0, 0.3], 'bar':[1,0.55], 'qux': [0.3,4.1]}
In [13]: df = pd.DataFrame.from_dict(mydict, orient='index')
In [14]: df
Out[14]:
0 1
qux 0.3 4.10
foo 0.0 0.30
bar 1.0 0.55
我想要做的是用0替换所有小于1的值。 产量:
0 1
qux 0 4.10
foo 0 0
bar 1.0 0
我怎样才能做到这一点?
答案 0 :(得分:11)
使用布尔索引并传递条件:
In [155]:
df[df<1] = 0
df
Out[155]:
0 1
bar 1 0.0
foo 0 0.0
qux 0 4.1
只是为了显示这里发生的事情,执行df < 1
将返回一个布尔索引:
In [156]:
df < 1
Out[156]:
0 1
bar False True
foo True True
qux True False
然后,我们将df
作为掩码传递给df[df<1]
,然后将新值指定为{{1}},请参阅docs以获取更多示例