我有一个DataFrame,如下所示:
df[16820:16830] data0 start_stop 16820 1 0 16821 1 1 16822 1 0 16823 1 0 16824 1 0 16825 1 -1 16826 0 0 16827 0 0 16828 1 1 16829 0 0 16830 1 -1
我需要做的是将start_stop列中1和-1之间的值标记为有效(1表示’start’,-1表示’stop’),-1和1之间的值作为无效(我稍后将丢弃的垃圾) .
有没有有效的方法来做这个而不是在整个数据帧上循环迭代?
最终结果如下:
data0 start_stop valid 16820 1 0 False 16821 1 1 True 16822 1 0 True 16823 1 0 True 16824 1 0 True 16825 1 -1 False 16826 0 0 False 16827 0 0 False 16828 1 1 True 16829 0 0 True 16830 1 -1 False ...
实现它的相关循环是,我认为:
df = df.reset_index(drop=True) value = False for i in range(0,df.shape[0]): if df.loc[i,'start_stop'] == 1: df.loc[i,'valid'] = True value = True elif df.loc[i,'start_stop'] == -1: df.loc[i,'valid'] = False value = False if df.loc[i,'start_stop'] == 0: df.loc[i,'valid'] = value
谢谢!
解决方法
这应该工作
df['valid'] = df.start_stop.cumsum()
然后
df['valid'] = df['valid'].apply(lambda x: True if x==1 else False) df start_stop valid 0 0 False 1 1 True 2 0 True 3 0 True 4 0 True 5 -1 False 6 0 False 7 0 False 8 1 True 9 0 True 10 -1 False
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。