使用熊猫使用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