如何在python中找到每周前10名的表现值？

Question

我想在我的 DataFrame 中返回每周表现最好的 10 个（按平均计算）变量。大约是 2 年的数据

我正在使用 Python 来解决这个问题，但最终也想用 SQL 来解决。

我已经能够生成返回最近一周前 10 名的代码，但希望每周都有结果

• 创建创建日期时间范围的 df

range_max = rtbinds['pricedate'].max()
range_min = range_max - datetime.timedelta(days=7)

sliced_df = rtbinds[(rtbinds['pricedate'] >= range_min) 
                    & (rtbinds['pricedate'] <= range_max)]

• 按“影子”分组

sliced_df.groupby(['pricedate','cons_name']).aggregate(np.mean)
         .sort_values('shadow').head(10)

• 返回第一周的数据。

    pricedate   cons_name       shadow
    2019-04-26  TEMP71_24753    -643.691
    2019-04-27  TMP175_24736    -508.062
    2019-04-25  TMP109_22593    -383.263
    2019-04-23  TEMP48_24759    -376.967
    2019-04-29  TEMP71_24753    -356.476
                TMP175_24736    -327.230
                TMP273_23483    -303.234
    2019-04-27  TEMP71_24753    -294.377
    2019-04-28  TMP175_24736    -272.603
                TMP109_22593    -270.887

但是，我想要一个返回每周前 10 名的列表，直到我的数据的最早日期

Answer 1

注意pd.sort_values 默认按升序排序，所以当您使用head(10) 时，如果我们考虑实数的自然排序，它实际上是最差的 10 个。

现在解决您的问题，这里有一个解决方案

首先我们需要创建一些列来标识一年中的第几周（rtbins 被重命名为 df）：

df['year'] = df['pricedate'].apply(lambda x: x.year)
df['week'] = df['pricedate'].apply(lambda x: x.isocalendar()[1])

然后我们将数据按['year', 'week', 'cons_name']分组：

df2 = df.groupby(['year', 'week', 'cons_name'], as_index=False).aggregate(np.mean)

您现在应该获得一个数据框，其中对于每个（年、周）您只有一条带有平均阴影的 cons_name 记录。

然后我们将每个（年、周）抽取前 10 名

def udf(df):
    return df.sort_values('shadow').head(10)

df2.groupby(['year', 'week'], as_index=False).apply(udf)

这应该会给你想要的结果。