如何通过聚合循环运行数据帧列表?

时间:2019-02-26 15:50:38

标签: python pandas loops datetime aggregate

下面的代码块可与单个数据帧完美配合,它采用日期时间序列数据帧,并为四个传感器列创建平均1小时的滚动滞后窗口。我有一个要执行此操作的数据帧列表,是否可以循环浏览列表或创建一个函数来执行此操作,所以我没有重复的代码块?

数据帧列表:

df_list = [df_t6,
           df_t7,
           df_t8,
           df_t11,
           df_t14,
           df_t15,
           df_t17,
           df_t19]

按原样工作的代码块:

# df_t6 telemetry means lag window

# create an empty list 'temp'
temp = [] 
# define the feature columns to be iterated
features = ['HP', 'Coolant1', 'AccumulatedWork', 'CuttingHP']
# loop
for column in features:
    # append to the list 'temp' a three hour (1H) sample taking the mean for each 'column' from the 'features' list
    temp.append(pd.pivot_table(df_t6, index = 'datetime', columns = 'Tool', values = column)
                .resample('1H', closed = 'left', label = 'right').mean().unstack())
# create a dataframe to hold the information and concat the 'temp' list
sensorData1H_mean = pd.concat(temp, axis = 1)
# name the columns using the list 'features' + '1H_mean'
sensorData1H_mean.columns = [n + '_1H_mean' for n in features]
# reset the index values
sensorData1H_mean.reset_index(inplace = True)

我知道我可以为此定义一种方法来快速迭代,但是我想知道是否有更快/更好的方法?

def oneHmean(d):
    # create an empty list 'temp'
    temp = [] 
    # define the feature columns to be iterated
    features = ['HP', 'Coolant1', 'AccumulatedWork', 'CuttingHP']
    # loop
    for column in features:
        # append to the list 'temp' a three hour (1H) sample taking the mean for each 'column' from the 'features' list
        temp.append(pd.pivot_table(d, index = 'datetime', columns = 'Tool', values = column)
                    .resample('1H', closed = 'left', label = 'right').mean().unstack())
    # create a dataframe to hold the information and concat the 'temp' list
    sensorData1H_mean = pd.concat(temp, axis = 1)
    # name the columns using the list 'features' + '1H_mean'
    sensorData1H_mean.columns = [n + '_1H_mean' for n in features]
    # reset the index values
    sensorData1H_mean.reset_index(inplace = True)
    return sensorData1H_mean

df_t6_m = oneHmean(df_t6)
df_t7_m = oneHmean(df_t7)

等...

子集:

df_t6:

   Unnamed: 0  IDData    HP  Coolant1  AccumulatedWork  CuttingHP Tool          datetime
0           0       0     0       388            30452      -1775   T6   2019-02-22 11:50:21 
1           1       1  1812       388            30452         37   T6   2019-02-22 11:50:21 
2           2       2  1775       388            30452          0   T6   2019-02-22 11:50:21
3           3       3  1797       382            30452         22   T6   2019-02-22 11:50:21
4           4       4  1797       382            30452         22   T6   2019-02-22 11:50:21 

df_t7:

   Unnamed: 0  IDData    HP  Coolant1  AccumulatedWork  CuttingHP Tool          datetime  
0           0       0  1646        14             3291      -1912   T7   2019-02-22 11:50:42
1           1       1  1680        14             3291      -1878   T7   2019-02-22 11:50:42 
2           2       2  1719        14             3291      -1839   T7   2019-02-22 11:50:42  
3           3       3  1673        14             3291      -1885   T7   2019-02-22 11:50:42
4           4       4  1648        14             3291      -1910   T7   2019-02-22 11:50:42 

1 个答案:

答案 0 :(得分:1)

我认为您可能希望将dfgroupbykey连接起来,然后应用oneHmean函数。

# concat the dfs into one, add a key for each to separate them
df = pd.concat([
    df_t6,
    df_t7
], keys=[
    't6', 't7'
])

# your function
def oneHmean(d):
    # create an empty list 'temp'
    temp = [] 
    # define the feature columns to be iterated
    features = ['HP', 'Coolant1', 'AccumulatedWork', 'CuttingHP']
    # loop
    for column in features:
        # append to the list 'temp' a three hour (1H) sample taking the mean for each 'column' from the 'features' list
        temp.append(pd.pivot_table(d, index = 'datetime', columns = 'Tool', values = column)
                    .resample('1H', closed = 'left', label = 'right').mean().unstack())
    # create a dataframe to hold the information and concat the 'temp' list
    sensorData1H_mean = pd.concat(temp, axis = 1)
    # name the columns using the list 'features' + '1H_mean'
    sensorData1H_mean.columns = [n + '_1H_mean' for n in features]
    # reset the index values
    sensorData1H_mean.reset_index(inplace = True)
    return sensorData1H_mean

# group on the keys and apply your function
df.groupby(level=0).apply(oneHmean)

结果

results