python - 遍历数据框的行并按组重新分配最小值

Question

我正在使用如下所示的数据框。

  id time diff
0 0   34   nan
1 0   36   2
2 1   43   7
3 1   55   12
4 1   59   4
5 2   2    -57
6 2   10   8

什么是通过 id 找到“time”的最小值，然后在这些最小值处将“diff”设置为 nan 的有效方法。我正在寻找一种解决方案，结果是:

   id time diff
0 0   34   nan
1 0   36   2
2 1   43   nan
3 1   55   12
4 1   59   4
5 2   2    nan
6 2   10   8

Answer 1

groupby('id') 并使用 idxmin 找到 'time' 的最小值的位置。最后使用loc赋值给np.nan

df.loc[df.groupby('id').time.idxmin(), 'diff'] = np.nan
df