如何用Pandas简洁删除tag为train_loop且start为NaN的行?
更简洁的方式删除指定条件的DataFrame行
你的现有代码可以正常工作,但借助pandas的矢量化操作,我们可以写出更简洁高效的实现,无需额外定义过滤函数:
方法一:布尔索引直接组合条件
old_len = len(df) df = df[~((df["tag"] == "train_loop") & df["start"].isna())]
方法二:结合drop与query方法
old_len = len(df) df = df.drop(df.query('tag == "train_loop" and start.isna()').index)
这两种写法的优势在于:
- 代码更紧凑,无需额外定义
filter_fn函数 - 采用pandas原生的矢量化计算,比逐行调用
apply的效率更高,尤其在处理大型数据集时差距会更明显 - 逻辑直观,直接通过条件组合表达筛选规则,可读性更强
内容的提问来源于stack exchange,提问作者Foobar
相关产品推荐
相关产品推荐

