python - 基于三个现有列创建新列

Question

我有一个包含三列、target_degrees、low_degrees和的数据框high_degrees。我想创建一个标记为成功的新列，用于检查 target_degrees 是否位于low_degrees和之间high_degrees。

示例数据框：

target_degrees   low_degrees   high_degrees   success
   10                0              50          1
   50                45             100         1
   20               100             200         0
   1                300             350         0

我尝试np.where在以下代码中使用，但出现语法错误。

df['success'] = np.where(df['target_degrees'] is in np.arange(df['low_degrees'], df['high_degrees']), 1, 0)

标签： pythonpython-3.xpandasnumpydataframe

Answer 1

使用多个条件：

df['success'] = np.where(((df['target_degrees'] >= df['low_degrees']) & (df['target_degrees']<= df['high_degrees'])), 1, 0)

输出：

  target_degrees  low_degrees   high_degrees    success
0       10                 0         50            1
1       50                45        100            1
2       20               100        200            0
3        1               300        350            0