Условное вычисление столбцов Pandas Dataframe

Question

У меня есть pandas dataframe, который читает

Category  Sales  
A           10
B           20

Я хочу сделать условное создание новой цели столбца

И я хочу, чтобы моя цель df выглядела как

Category  Sales  Target 
A           10    5
B           20   10

Я использовал приведенный ниже код, и он выдал ошибку

if(df['Category']=='A'):
    df['Target']=df['Sales']-5
else:
    df['Target']=df['Sales']-10

Answer 1

Использовать векторизацию numpy.where:

df['Target']= np.where(df['Category']=='A', df['Sales'] - 5, df['Sales'] - 10)
print (df)
  Category  Sales  Target
0        A     10       5
1        B     20      10