微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

使用熊猫使用if条件

如何解决使用熊猫使用if条件

我在几分钟之内就有start_timeend_timetime_taken之类的列。 我想仅在time_taken等于10分钟时生成具有完整数据的csv文件

我当前的代码是:

if ( df['time_taken']) == 10:
  df.to_csv(r'result.csv')

出现类似以下错误

The truth value of a Series is ambiguous. Use a.empty,a.bool(),a.item(),a.any() or a.all().

解决方法

您可以在保存之前对数据框进行子集设置,从而仅保存符合条件的数据框部分:

df.loc[df.time_taken == 10].to_csv("results.csv")

您可能想阅读.loc

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。