在多索引數據幀中查找每個索引的第一次出現

Question

我有一個名為 panel_long 的熊貓數據框，看起來像這樣：

                 x  y  z event Start
company period   
A       1        1  2  3 False 5
B       1        1  2  3 False 7
...
A       27       1  2  3 True  5
B       27       1  2  3 False 7
...
A       31       1  2  3 False 5
B       31       1  2  3 True  7

至於與生存分析的准備livelines我想補充，其中列明了第一期活動相當於每家公司真正的一列。 對於非多索引數據幀，我會使用 idxmax，但我不知道如何使它適用於多索引 df。 其中，我嘗試了以下方法：

def funct(df, column):
    df[column].idxmax()

df = panel_long.groupby('company').apply(funct, ('event'))

但這只是返回一個空的數據框。 有任何想法嗎？ 我被卡住了

Answer 1

您應該在自定義函數中return輸出

def funct(df, column):
    return df[column].idxmax()
df.groupby('company').apply(funct, ('event'))
Out[248]: 
company
A    (A, 1)
B    (B, 1)
dtype: object