python - 从 pandas 的数据框中获取第 1 列中的值 = 指定的第 2 列的最大值

Question

我有一个 pandas 数据框。例如:

df=
  paper id  year
0         3  1997
1         3  1999
2         3  1999
3         3  1999
4         6  1997
                so on

我想要与作为输入给出的论文 ID 相对应的最大年份。例如，如果给定的论文 ID 为 3，我希望将 1999 作为答案。

我该怎么做？

Answer 1

有 2 个通用解决方案 - 先过滤，然后获取 max:

s = df.loc[df['paper id'] == 3, 'year'].max()
print (s)
1999

<小时/>

s = df.set_index('paper id').loc[3, 'year'].max()
print (s)
1999

或者将 max 聚合到 Series，然后按 index 值进行选择:

s = df.groupby('paper id')['year'].max()
print (s)
paper id
3    1999
6    1997
Name: year, dtype: int64

print (s[3])
1999