python - Pandas - 将值转换为 1 和 0 的快速方法

Question

我有一个数据框“df”:

   x
0 -2
1  2
2 -4
3  2
4  9
5 -2

我正在尝试找到一种快速创建新列 df['binary'] 的方法，其中 df['x'] 中的所有正值都输入为 1，而 df['x'] 中的负值是在新的 df['binary'] 列中输入为 0。真实的框架非常大，所以我试图快速完成。任何帮助表示赞赏。

谢谢

Answer 1

赋值 + gt

df.assign(binary=df.x.gt(0) * 1)

   x  binary
0 -2       0
1  2       1
2 -4       0
3  2       1
4  9       1
5 -2       0

---

如果速度是你的追求

df['binary'] = (df.x.values > 0).astype(np.uint8)