我有一个名为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'))
但这只是返回一个空的数据框。有任何想法吗？我被卡住了
您应该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
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句