python - 根据 pandas 中的条件创建 bool 列

Question

我有一个数据框，如下所示

ID    DURATION   STATUS   CONSIDER
1     30         ACTIVE   True
2     780        CLOSED   True
3     745        ACTIVE   False
4     366        ACTIVE   False
5     367        ACTIVE   True

我想创建一个新列 ESTIMATION_FLAG，当 DURATION < 700 且 STATUS == ACTIVE 且 CONSIDER == True 时，该列为 True。

预期输出:

ID    DURATION   STATUS   CONSIDER  ESTIMATION_FLAG
1     30         ACTIVE   True      True
2     780        CLOSED   True      False
3     745        ACTIVE   False     False
4     366        ACTIVE   False     False
5     367        ACTIVE   True      True

Answer 1

使用Series.lt和 Series.eq对于掩码和链式 & 对于按位 AND，优点是较少 ():

df['ESTIMATION_FLAG'] = df['DURATION'].lt(700) & df['STATUS'].eq('ACTIVE') & df['CONSIDER']
print (df)
   ID  DURATION  STATUS  CONSIDER  ESTIMATION_FLAG
0   1        30  ACTIVE      True             True
1   2       780  CLOSED      True            False
2   3       745  ACTIVE     False            False
3   4       366  ACTIVE     False            False
4   5       367  ACTIVE      True             True