Question

这类似于我之前提出的问题，但由于解决方案在数据分组时不起作用，因此差异很大：

给出一些数据：

import pandas as pd
import numpy as np
import datetime

data = {'group':['a', 'a', 'a','b','a', 'b'], 
        'value': [1,2,3,4,3,5], 'names': ['joe', 'bob', 'greg','joe', 'bob', 'greg'],
        'dates': ['2015-01-01', '2015-01-02', '2015-01-03', '2015-01-03', '2015-01-04', '2015-01-04']}

df = pd.DataFrame(data=data, columns=["group", "value", "names"], 
                  index=pd.to_datetime(data['dates']))

给出：

           group value names
2015-01-01     a     1   joe
2015-01-02     a     2   bob
2015-01-03     a     3  greg
2015-01-03     b     4   joe
2015-01-04     a     3   bob
2015-01-04     b     5  greg

我希望得到：

            group value names
2015-01-01      a     2   bob
2015-01-03      a     3   bob
2015-01-03      b     5  greg

因此数据被分组，重新采样2天（'2D'），然后收集对应于最大'值'的名称我尝试了下面的错误：

(df.groupby('group').resample('2D')[['value']].idxmax()
   .assign(names=lambda x: df.loc[x.value]['names'].values,
           value=lambda x: df.loc[x.value]['value'].values)
)

Answer 1

您可以在分组后使用apply对值进行排序，按值命名列，然后选择第一行。

g = df.groupby(['group', pd.Grouper(freq='2D')])[['value', 'names']]
g.apply(lambda x: x.sort_values(['value', 'names'], ascending=[False, True]).iloc[0])\
 .reset_index('group')

           group  value names
2015-01-01     a      2   bob
2015-01-03     a      3   bob
2015-01-03     b      5  greg

这与使用resample

相同

g = df.groupby(['group'])[['value', 'names']]
g.resample('2D').apply(lambda x: x.sort_values(['value', 'names'], ascending=[False, True]).iloc[0])\
 .reset_index('group')

Pandas：重新采样分组的数据帧列，获得与最大值对应的离散特征

1 个答案: