我正在尝试对数据帧执行特定操作.
给出以下数据帧：
df1 = pd.DataFrame({
'id': [0, 1, 2, 1, 3, 0],
'letter': ['a','b','c','b','b','a'],
'status':[0,1,0,0,0,1]})
id letter status
0 a 0
1 b 1
2 c 0
1 b 0
3 b 0
0 a 1
我想基于以下限制创建另一个包含来自df1的行的数据帧.
如果2行或更多行具有相同的id和字母,则返回状态为1的行.必须复制所有其他行.
生成的数据框应如下所示：
id letter status
0 a 1
1 b 1
2 c 0
3 b 0
任何帮助是极大的赞赏.谢谢
解决方法:
这应该工作：
>>> fn = lambda obj: obj[obj.status == 1] if any(obj.status == 1) else obj
>>> df.groupby(['id', 'letter'], as_index=False).apply(fn)
id letter status
5 0 a 1
1 1 b 1
2 2 c 0
4 3 b 0
[4 rows x 3 columns]
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 [email protected] 举报，一经查实，本站将立刻删除。
