用大熊貓的固定開始日期計算滾動中位數

Question

我有一個帶有以下.head（）的pandas DataFrame：

       EC       Date       PE
0  C00529 2000-01-31  21.8799
1  C00529 2000-02-29  24.4603
2  C00529 2000-03-31  17.2053
3  C00529 2000-04-30  17.5083
4  C00529 2000-05-31  18.0368

EC是公司的ID。 我想為每家公司開列一個中位數為“ PE”的列，但是從最小日期（“日期”）開始滾動。 知道如何在熊貓中這樣做嗎？

謝謝！

Answer 1

您可以使用df.expanding方法（ 請參見此處 ）。 假設您的DataFrame已排序：

df.expanding()['PE'].median()

假設它沒有排序：

df.loc[df['Date'].sort_values().index].expanding()['PE'].median()