我是熊猫的新手.我有一个像这样的Pandas数据框:
df = pd.DataFrame(data={'id': [1,1,2,2],'val1': [0,0]})
我想添加一个列val2,它指示一行是否落在另一行具有与其自身相同的id的行中,其中val1 == 1.
结果将是一个数据框,如:
df = pd.DataFrame(data={'id': [1,0],'val2': [0,1]})
我的第一个想法是使用apply语句,但这些只是按行进行.从我的循环经验来看,永远不是答案.任何帮助将不胜感激!
解决方法
让我们尝试在groupby中移动cumsum.
df['val2'] = df.groupby('id').val1.apply( lambda x: x.shift().cumsum() ).ge(1).astype(int)
或者,为了避免lambda,
df['val2'] = ( df.groupby('id') .val1.shift() .groupby(df.id) .cumsum() .ge(1) .astype(int) )
df id val1 val2 0 1 0 0 1 1 1 0 2 1 0 1 3 2 0 0 4 2 1 0 5 2 0 1 6 2 0 1
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。