我有一个定义如下的熊猫系列:
import pandas as pd
series = pd.Series([11,9,13], index=[1,2,3])
我使用以下方法计算滚动序列:
rolling = series.rolling(window=3, min_periods=3, center=True)
我现在想将给定的
method
应用于滚动的series
,如下所示:method = 'median'
if method == 'median':
series = rolling.median()
elif method == 'mean':
series = rolling.mean()
可以仅使用字符串名称来实现吗?例如,我会做类似的事情:
if method in ['mean', 'median']:
series = rolling.method()
感谢您的帮助。
最佳答案
是的,您可以使用内置的getattr
函数。
if method in ('mean', 'median'):
series = getattr(rolling, method)()
万一看起来很阴暗:
>>> getattr(rolling, median) == rolling.median
True