如何用条件替换 Pandas 数据框中所有列中的所有值

Question

我有以下数据框：

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

我想要做的是将所有小于 1 的值替换为 0。 产量：

       0     1
qux  0     4.10
foo  0     0
bar  1.0   0

我怎样才能做到这一点？

Answer 1

使用布尔索引并传递条件：

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]，更多示例请参见docs