分类数据：如果值是 x 则 y

Question

我在数据框中有一列这样的值：

   0

0   1.047297

1   1.038211

2   1.035080

3   1.005149

我想创建一个新列，如果该列的一个 'cell' 中的值大于 x，那么它将 return 1 并且如果没有，则 0.

本质上我是在尝试模仿一个 excel 函数：if(cell > x,1,0)

这是一项非常简单的任务，但我找不到简单的答案 - 将不胜感激！

Answer 1

import numpy as np
df = # your data frame
df['newcolumn'] = np.where(df['cell'] > x, 1, 0)

有关详细信息，请参阅 numpy.where 文档。