将值转换为数据框或特定列中阈值的虚拟变量

Question

我想在一行中或使用 pandas 内置函数在下面重现此确切代码：

您有一个包含 N 列值的数据框，您希望为正值分配 1，为负值分配 0

mask_dummies_1 = df > 0    # Create a mask for positive values
mask_dummies_0 = df < 0    # Create a mask for negative values
df[mask_dummies_1] = 1     # apply positive mask and take the value 1
df[mask_dummies_0] = 0     # apply positive mask and take the value 0

我知道那很脏^^' 有什么猜测吗？

Answer 1

解决方案，一条线，显式：

df = 1 * (df > 0)

一栏：

df['col'] = 1 * (df['col'] > 0)