要应用于GroupBy的写入函数

Question

从这篇文章开始：Returning subset of each group from a pandas groupby object我正在尝试编写一个函数来应用于pandas中的groupby对象

def group_by_function(df):
    ID = df.loc[df.Parameter_1==0].Parameter_2.idxmin()        
    df_2 = df.iloc[ID].Parameter

    print(ID)
    return df_2

df.groupby(by=['Column1', 'Column2']).apply(group_by_function)

我对索引在这种情况下的工作方式有点迷茫。在这个例子中，我返回的ID是1189-但是行是我做的测向iloc[1189]返回位置指示器超出界限的错误。你知道吗

我的理解是，索引应该在groupby期间保留，这是ID=line告诉我的。但是我不清楚为什么iloc调用会抛出一个错误。你知道吗

我的groupby是由两列组成的-不确定这是否是一个因素。你知道吗

本

Answer 1

可以使用loc按标签选择，因为idxmin返回索引，而不是位置：

df_2 = df.loc[ID, 'Parameter']