相当于numpy where函数在熊猫中

Kben59

这是我的问题,我有两列如下:

 A   B
-3   4
-6  -8
12  14

我想创建一个代码,如果值> -2则将所有值替换为1,否则将值替换为0。

我知道使用numpy会很简单,d["newcolum"]=np.where(d < -2,1,0)但是使用pandas会遇到更多的困难,因为它的where函数仅在不满足条件时才显示值。

你们知道有什么可以帮助我的吗?谢谢

耶斯列尔

如果要设置01将值转换为布尔掩码为整数:

df = (df < -2).astype(int)
print (df)
   A  B
0  1  0
1  1  1
2  0  0

使用np.whereia绝对好,只需要DataFrame构造函数即可:

df = pd.DataFrame(np.where(df < -2,1,0), index=df.index, columns=df.columns)
print (df)
   A  B
0  1  0
1  1  1
2  0  0

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章