Как сделать расширение функций с помощью панд на временных рядов

Question

В идеале я хотел бы, чтобы в конечном итоге, как это (обратите внимание на modelIDs сгруппированных и expanding_mean, expanding_median функции, SaleCount и т.д. на сегодняшний день):

Я попытался сортировать данные и переиндексировать «saledate», а затем использовать «group_by» для выполнения функций для идентификатора модели, но, похоже, не может получить его в желаемом формате. Я работаю над тем, что кажется, должно быть довольно простой задачей в течение 3 дней прямо сейчас. Любые предложения будут ценны.

Answer 1

Попробуйте использовать transform и shift объединены вместе:

from pandas.stats.moments import * 

df.groupby(['ModelID', 'saledate']).transform(lambda x: expanding_mean(x.shift(1)))