主页 / 问题 / 1404685
Accepted
leonardik
Asked:2022-06-28 20:26:54 +0800 CST2022-06-28 20:26:54 +0800 CST

以给定的步骤滚动

  • 772

大家好！有一个脚本使用滚动来查找列中的最大值。当前窗口步长 = 1，即(1,3,1)下一个滑动序列(3,1,5)等。

如何设置 step = 3，即序列为(1,3,1)and (5,10,5)？

屏幕截图显示了窗口的移动，脚本最终应显示的最大值以红色突出显示。

在此处输入图像描述

找到了解决方案df.rolling(window=3,step=3)，但它不起作用......

import pandas as pd
sample = {
    'ID':[1,1,1,9,9,9],
    'Quant' :[1,3,1,5,10,5]
              }
df = pd.DataFrame(sample)

df2=df.rolling(3).agg({'Quant':'max'})
df3=df2.dropna(how='all')
print(df3)
python pandas

2 个回答

  1. Best Answer

    您误解了滚动方法的工作原理。从任务的描述来看，它根本不适合你。我建议这样做：

    import pandas as pd
import numpy as np
sample = {
    'ID':[1,1,1,9,9,9],
    'Quant' :[1,3,1,5,10,5]
              }
df = pd.DataFrame(sample)
step = 3

res = [x.Quant.max() for x in np.array_split(df, len(df)//step)]
print(res)

    [3, 10]

    也就是说，在np.array_split您的帮助下将数据框分成所需长度的片段，然后在每个片段中查找最大列Quant。

    • 3

  2. 您还可以获得所需的最大值：

    res = [df.iloc[i:i+3,1].max() for i in range(0,len(df),3)]
'''
[3, 10]

    或过滤您想要的行：

    res = df.iloc[[df.iloc[i:i+3,1].idxmax() for i in range(0,len(df),3)]]
'''
   ID  Quant
1   1      3
4   9     10
    • 2

相关问题