回答編集履歴

1

2023/05/28 04:53

投稿

melian
melian

スコア19825

test CHANGED
@@ -9,7 +9,7 @@
9
9
  #
10
10
  N = 3
11
11
  df_gr = df.groupby('product').head(N).groupby('product')\
12
- .agg({'price': lambda x: x.tail(1) if len(x.iloc[N-1:]) else np.nan})\
12
+ .agg({'price': lambda x: x.tail(1) if len(x) >= N else np.nan})\
13
13
  .reset_index()
14
14
 
15
15
  print(df_gr)