问题描述
我在几分钟之内就有start_time
,end_time
和time_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