时间戳

首页 > 解决方案 > 如果不存在，Pandas 通过查找最小列进行分组返回 NaN

python - 如果不存在，Pandas 通过查找最小列进行分组返回 NaN

问题描述

假设我有以下数据框：

import pandas as pd
df = pd.DataFrame({'id': [1,1,1,2,3,2], 'year': ['2020', '2014', '2002', '2020', '2016', '2014'], 'e': [True, False, True, True, False, True]})

df.info()
id  year      e
  1  2020   True
  1  2014  False
  1  2002   True
  2  2020   True
  3  2016  False
  2  2014   True

而且我想找到每个id的最小年份，其中e为True，如果e中没有任何True，则该id返回NaN。最终结果将是：

id  year
1   2002
2   2014
3   NaN

标签： pythonpandasdataframe

解决方案

尝试groupby前后reindex过滤

s = df.loc[df.e].groupby('id').year.min().reindex(df.id.unique()).reset_index()

s
Out[307]: 
   id  year
0   1  2002
1   2  2014
2   3   NaN

或转换为Categorical

df['id'] = pd.Categorical(df['id'])
df.loc[df.e].groupby('id').year.min()
Out[309]: 
id
1    2002
2    2014
3    None
Name: year, dtype: object

推荐阅读