从 Python Pandas 中的第一行计算值的函数

Question

pandas 中是否有任何函数可以模拟 '=sum($A$1:A10'(for 10th row) 之类的 excel 公式，即公式应该从第一行获取滚动数据。

Pandas 滚动函数需要一个整数值作为窗口参数。

Answer 1

pandas 中=SUM($A$1:A1) 的等价物是.expanding().sum()（需要pandas 0.18.0）：

ser = pd.Series([1, 2, 3, 4])

ser
Out[3]: 
0    1
1    2
2    3
3    4
dtype: int64

ser.expanding().sum()
Out[4]: 
0     1.0
1     3.0
2     6.0
3    10.0

您还可以通过 apply 应用泛型函数：

ser.expanding().apply(lambda x: np.percentile(x, 90))
Out: 
0    1.0
1    1.9
2    2.8
3    3.7
dtype: float64

或者直接用分位数：

ser.expanding().quantile(0.9)
Out[15]: 
0    1.0
1    1.0
2    2.0
3    3.0
dtype: float64

请注意，第 90 个百分位数等于第 0.9 个分位数。然而，Series.quantile 和 Series.expanding.quantile 返回不同的结果，可能是a bug。

np.percentile 返回与 Excel 的 PERCENTILE.INC 相同的结果。对于PERCENTILE.EXC，我之前写过一个小函数here。