Pandas 滚动 window 等级顺序与条件

Question

我想我有 Pandas DataFrame 如下所示：

  account   amount  
  A         10      
  A         10      
  A         20      
  A         10      
  A         10      
  A         10      
  B         10      
  B         10      
  B         20      
  B         18      
  B         18      
  B         12      
  B         14      
  B         18      
  B         19      

我想要一个如下所示的新列：

  account   want  
  A         nan   
  A         nan   
  A         1     
  A         2     
  A         3     
  A         1     
  B         nan   
  B         nan   
  B         1     
  B         2     
  B         3     
  B         1     
  B         2     
  B         1     
  B         1     

这个想法是在给定 window 中找到最大值之后对顺序（从 1 开始）进行排序。

例如，给定的 window 为3 ，在账户A中，最大金额为20 。 然后，索引2将返回1 ，因为20是给定 window 中的最大值。 之后，对于下一个滚动 window，在索引3处将返回2 ，因为数量为10 ，比上面的20少。

将逻辑应用于帐户B ，结果如图所示。 我可以考虑用 3 创建最大的滚动 window 数量，但我不能再做进一步了。

Answer 1

我想我可以通过下面的代码解决这个问题：

def rank(window):
    newWindow = np.array(window)
    lastMax = np.where(newWindow == np.amax(newWindow))[0][-1]
    maxLocation = np.repeat(lastMax, newWindow.shape[0])
    location = np.arange(0, newWindow.shape[0])    
    ranks = location - maxLocation + 1
    return ranks[-1]

df['want'] = df.groupby('account')['amount'].rolling(window = 3).apply(rank).droplevel(0)

感谢大家！