根据条件转换熊猫数据框列

mom*_*ind 3 python pandas

我有一个值从 0.0 到 1.0 的 pandas 列。

我想根据阈值将此列转换为二进制列(0 或 1),即,如果值 <= 阈值,否则它将变为 0 和 1。

jez*_*ael 5

通过gt( >)创建布尔掩码,然后将其转换为integers:

df = pd.DataFrame({'col':[.4,0.5,.1]})

threshold = .2
df['new'] = df['col'].gt(threshold).astype(int)
print (df)
   col  new
0  0.4    1
1  0.5    1
2  0.1    0
Run Code Online (Sandbox Code Playgroud)