在Pandas中标记数据

Question

我有一个很大的数据帧。我想像这样对我的数据进行分类：

if element > threshold:
    element = 1
elif element < -threshold:
    element = -1
else:
    element = 0

尝试在熊猫中这样做，但它显然省略了范围[-threshold，threshold]。有没有办法在整个列上使用“else”？

for col in data:
    data[col][data[col] > threshold] = 1
    data[col][data[col] < -threshold] = -1

Answer 1

使用np.sign

np.sign(data - threshold)