Python Pandas groupby函数,具有每月、每小时平均值的动态参数



几年来,我有一个以下格式的每小时数据帧:

Date/Time            Value
01.03.2010 00:00:00  60
01.03.2010 01:00:00  50
01.03.2010 02:00:00  52
01.03.2010 03:00:00  49
.
.
.
31.12.2013 23:00:00  77

我使用以下代码来获得数据中每年每小时的平均值:

In [11]: year_hour_means = df1.groupby(lambda x: (x.year, x.hour)).mean()
In [12]: year_hour_means
Out[12]:
           Value
(2010, 0)     60
(2010, 1)     50
(2010, 2)     52
(2010, 3)     49

现在我想把这些代码放入一个函数中,这样我就可以动态地选择按季度、年或月对小时进行分组,也可以对数据帧的特定日期范围进行分组。

我已经编写了以下函数:

def datameans(df, avggrouper1, avggrouper2, startdate, enddate):  
    import pandas as pd
    df_hour_means = df[startdate:enddate]      
    df_hour_means = df_hour_means.groupby(lambda x: (avggrouper1, avggrouper2)).mean()  
    print df_hour_means.to_string()
    df_hour_means.plot()
    pass

我调用类似的函数

datameans(dataframe, 'x.quarter', 'x.hour' , '2010-01-01 00:00:00', '2012-12-31 23:00:00')

不幸的是,这不起作用。有人能帮助我如何将年份、季度、月份和天数作为不同的参数来计算平均值吗?

我认为您正在寻找的是getattr:

def datameans(df, avggrouper1, avggrouper2, startdate, enddate):  
    df_hour_means = df[startdate:enddate]      
    df_hour_means = df_hour_means.groupby(
        lambda x: (getattr(x,avggrouper1), getattr(x,avggrouper2))).mean()  
    print df_hour_means.to_string()
    df_hour_means.plot()

(就像Matti John的回答一样)你会用呼叫datameans

datameans(dataframe, 'quarter', 'hour' , '2010-01-01 00:00:00', '2012-12-31 23:00:00')

或者,使用operator.attrgetter:

import operator
keyfunc = operator.attrgetter(avggrouper1, avggrouper2)
df_hour_means = df_hour_means.groupby(keyfunc).mean()  

最新更新