时间戳

首页 > 解决方案 > 使用 pandas 使用 if 条件生成 csv 文件

python - 使用 pandas 使用 if 条件生成 csv 文件

问题描述

我有像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().

标签： pythonpandasif-statement

解决方案

您可以在保存之前对数据框进行子集化，从而仅保存与您的条件匹配的数据框部分：

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

您可能想继续阅读.loc

推荐阅读