Фрейм данных, выбирающий строки с заданными условиями и работающий

Question

у меня датафрейм выглядит так:

 import pandas as pd    
    df = pd.DataFrame({'AA': [1, 1, 2, 2], 'BB': ['C', 'D', 'C', 'D'], 'CC': [10,20,30,40], 'DD':[], 'EE':[]})

Теперь я хочу умножить значение в столбце «CC» на число 2, если «AA» = 1 и «BB» = «C». Например, первая строка будет соответствовать условиям, поэтому значение в столбце «CC», равное 10, будет умножено на 2, и выходные данные перейдут в ту же строку в столбце «DD».

У меня будут другие требования к другим парам «AA» и «BB», но это будет хорошим началом, если я смогу понять, как применять умножение к строкам, которые удовлетворяют условиям.

Большое спасибо.

Answer 1

m0 = df.AA == 1
m1 = df.BB == "C"

df.loc[m0 & m1, "DD"] = df.loc[m0 & m1, "CC"] * 2