基于其他分类列的条件创建新列

Question

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

Category   Value
A          10
B          22
A          2
C          30
B          23
B          4
C          8
C          24
A          9

我需要创建一个标志列，基于以下条件：

If the values of Category A is greater than or equal 5 then Flag=1, else 0
If the values of Category B is greater than or equal 20 then Flag=1, else 0
If the values of Category C is greater than or equal 25 then Flag=1, else 0

期望的输出如下所示。

Category   Value   Flag
A          10      1
B          22      1
A          2       0
C          30      1
B          23      1
B          4       0
C          8       0
C          24      0
A          9       1

我尝试了以下代码。

df_['Flag'] = np.where(df_[df_['Category']=='A']['Value']>=5, 1, 0)
df_['Flag'] = np.where(df_[df_['Category']=='B']['Value']>=20, 1, 0)
df_['Flag'] = np.where(df_[df_['Category']=='C']['Value']>=25, 1, 0)

- Danish

Answer 1

首先使用位运算符&进行按位与操作，然后再使用位运算符|进行按位或操作：

m1 = (df['Category']=='A') & (df['Value']>=5)
m2 = (df['Category']=='B') & (df['Value']>=20)
m3 = (df['Category']=='C') & (df['Value']>=25)

df['Flag'] = np.where(m1 | m2 | m3, 1, 0)
print (df)
  Category  Value  Flag
0        A     10     1
1        B     22     1
2        A      2     0
3        C     30     1
4        B     23     1
5        B      4     0
6        C      8     0
7        C     24     0
8        A      9     1

或者将True/False映射为1/0:

df['Flag'] = (m1 | m2 | m3).astype(int)

- jezrael