提取每个用户的最后一个problem_id

我有一个包含以下列的数据帧: ['user_id', 'problem_id', 'timestamp'] 基本上是谁在什么时候解决了问题。显然,有些用户解决了许多问题。

我想提取每个用户解决的最后一个问题。我的第一种方法是按user_id分组并获得最大值: df_s.groupby('user_id').max()[['problem_id']] ,但仔细查看之后,我意识到它只会返回用户所解决的最高词典排序问题。

很明显,我还可以对groupby聚合进行迭代,对数据帧进行排序,并解决第一个问题,但我希望有一个简单的语句。