由id – python汇总多个字符串列

发布时间：2020-12-20 12:14:17 所属栏目：Python 来源：网络整理

导读：我在 Python中获得了以下数据框： d = pd.DataFrame({'id': [1,1,2,3],'col1': ['normal','well','normal','normal'],'col2': ['bad','bad']}) 我想按id汇总,但如果没有别的东西(‘well’或’bad’),请保留除’normal’或’normal’以外的列字符串.如下： re

我在 Python中获得了以下数据框：

d = pd.DataFrame({'id': [1,1,2,3],'col1': ['normal','well','normal','normal'],'col2': ['bad','bad']})

我想按id汇总,但如果没有别的东西(‘well’或’bad’),请保留除’normal’或’normal’以外的列字符串.如下：

result = pd.DataFrame({'id': [1,'col1': ['well','bad']})

我正在考虑排序,然后使用groupby和.first但不确定如何在每列的顶部获得所需的级别.

解决方法

如果在 GroupBy.first之前没有NaNs值,则首先使用 replace NaNs：

d = d.replace('normal',np.nan).groupby('id').first().fillna('normal')
#alternative solution
d = d.mask(d == 'normal').groupby('id').first().fillna('normal')

print (d)
      col1    col2
id                
1     well     bad
2     well  normal
3   normal     bad

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!