Расчет условного скользящего среднего в DataFrame - PullRequest
0 голосов
/ 30 апреля 2020

Скажите, у меня есть Pandas DataFrame (df) со столбцами A и B, содержащими числа с плавающей запятой.

Я пытаюсь вычислить скользящее среднее значение 10 для значений в столбце A, только если значения в столбце B больше чем 10 и создайте новый столбец C для результата.

Я использую следующий код:

df['C']=df.where(df['B'] > 10)['A'].rolling(10).mean()

Однако я получаю столбец C, полный значений nan. Мне известно, что оператор .where дает значение, если True, и Nan, если False, но оператор mean () по умолчанию игнорирует значения nan, поэтому не могу понять, в чем проблема ...

Заранее спасибо!

...