游乐游手机版
首页/编程语言/文章详情

Pandas按列条件筛选生成新DataFrame

时间:2026-07-14 06:56
通过布尔索引筛选DataFrame中满足列值条件的行,之后可链式调用 drop()移除条件列,从而得到结构清晰的新DataFrame。注意缺失值默认被排除,列名需严格匹配。另外,也可使用 loc直接指定保留列,更为简洁高效。
本文介绍如何使用布尔索引高效筛选DataFrame中满足特定列值条件的行,并支持在操作后移除该条件列,从而生成结构清晰、易于分析的新DataFrame。

在数据清洗过程中,依据列值条件筛选行是最常见的操作之一。以实际场景为例:假设你有一个名为 df1 的DataFrame,其中包含一列 "purpose",你需要提取所有 purpose == "house" 的记录,并将其保存为新的DataFrame df2。如果希望同时移除用于筛选的列,让表格结构更简洁,可以使用链式调用 .drop() 方法一步完成。

核心代码仅一行,非常直观:

df2 = df1[df1["purpose"] == "house"].drop("purpose", axis=1)

代码说明

  • df1["purpose"] == "house" 会生成一个布尔型Series,标记每一行是否满足条件;
  • df1[...] 利用该布尔索引直接筛选出所有符合条件的行,保留所有原始列;
  • .drop("purpose", axis=1) 沿着列方向(axis=1)删除 "purpose" 列,最终 df2 仅包含其他特征字段。

⚠️ 注意事项

  • 如果 "purpose" 列中存在缺失值(NaN),使用 == "house" 判断时,NaN 会被自动判定为 False(不会引发错误),即这些行默认被排除。如果需要考虑空值,可改用 df1["purpose"].str.contains("house", na=False) 或结合 isna() 单独处理。
  • 列名必须严格匹配——大小写与空格均需一致。建议先通过 df1.columns.tolist() 确认列名的准确写法。
  • 如果仅需保留某几列,不一定非要先筛选再删除列。更推荐使用 .loc 直接指定要保留的列,代码更清晰:
    df2 = df1.loc[df1["purpose"] == "house", ["col1", "col2", "col3"]]

该方法简洁、向量化,性能表现优异,千万级以下的数据子集提取完全够用。可以说,它是Pandas数据清洗流程中最基础也是最关键的一环。

来源:https://www.php.cn/faq/2819240.html
上一篇Pandas中按条件分组并实现跨False边界的前向与后向填充方法 下一篇布尔序列分组与跨组前向后向填充方法
本站内容用于信息整理与展示,如有侵权或内容问题请及时联系处理。

相关推荐

补充同频道和同主题内容,方便继续浏览更多相关内容。

同类最新

继续查看同栏目最近更新的文章。

更多
Go微服务熔断后指数退避重试机制配置
编程语言 · 2026-07-14

Go微服务熔断后指数退避重试机制配置

熔断器打开后应进入半开状态,再对试探请求启用指数退避重试,避免无效重试。使用gobreaker控制请求准入,backoff控制试探间隔,并启用抖动防止脉冲流量。重试和熔断需分层,重试只针对临时错误,熔断统计重试后的最终结果。

Java多重上界通配符无法直接写入语法的根本原因
编程语言 · 2026-07-14

Java多重上界通配符无法直接写入语法的根本原因

Java通配符仅支持单一上界,如?extendsA,无法直接使用多重上界。多重上界(如TextendsA&B)仅适用于泛型类型参数声明,这是Java泛型设计中的语法限制,旨在简化类型系统。若需多约束,需通过类型参数间接实现。

Golang微服务中集成Argo实现GitOps持续发布
编程语言 · 2026-07-14

Golang微服务中集成Argo实现GitOps持续发布

Go微服务与ArgoCD边界清晰,Application路径指向manifests目录而非源码。镜像更新通过CI自动提交或argocd-image-updater实现,避免写死latest标签。readinessProbe需合理配置initialDelaySeconds与periodSeconds,确保同步顺畅。

Java中AbstractList的快速失败机制中并发修改检查方法的执行时机
编程语言 · 2026-07-14

Java中AbstractList的快速失败机制中并发修改检查方法的执行时机

在AbstractList迭代器中,每次调用next()、remove()、previous()、set()或add()时,都会先执行checkForComodification,通过比较modCount与expectedModCount检测并发修改,确保操作时视图一致性,防止状态错乱。

Python中statistics模块快速计算统计学中位数的方法与步骤
编程语言 · 2026-07-14

Python中statistics模块快速计算统计学中位数的方法与步骤

使用Python的statistics median()计算中位数需注意:不接受空列表,否则抛出StatisticsError异常;不自动过滤None或非数字值;传入大型生成器可能耗尽内存或导致性能下降。建议先过滤脏数据并转为列表,再计算,同时明确空数据时的处理策略。