如何在 Pandas 中仅对指定 ID 值去重(保留首次出现,删除后续重复)

本文介绍一种精准控制去重范围的方法:仅对 dataframe 中某一特定 id 值(如 id==1)删除其重复行,其余 id 值完全保留原样,不参与去重逻辑。

在实际数据处理中,我们常需“条件性去重”——并非全局去重,而是仅对满足某条件的子集执行去重操作。例如,给定如下 DataFrame:

import pandas as pd

df = pd.DataFrame({
    'ID': [1, 2, 3, 1, 2, 3],
    'Value': ['A', 'B', 'A', 'B', 'C', 'A']
})

目标是:仅当 ID == 1 时,删除其后续重复项(即保留第一个 ID=1 的行,移除第二个 ID=1 的行),而 ID=2 和 ID=3 的所有行均完整保留(即使它们自身存在重复,也不处理)

实现该逻辑的核心思路是:构造布尔掩码(boolean mask),精准定位“既是目标 ID 又是重复项”的行,再取反筛选

具体步骤如下:

  1. 识别目标行:df.ID.eq(1) → 返回 True 仅当 ID 等于 1;
  2. 识别所有 ID 列的重复行(按首次出现保留原则):df.duplicated(subset=['ID']) → 对 ID 列标记从第二次出现起的所有重复行为 True;
  3. 组合条件:用 &(逻辑与)连接二者,得到仅对 ID==1 的重复行生效的掩码;
  4. 反向筛选:用 ~mask 保留非匹配行,即:保留所有非 ID=1 的行 + 保留 ID=1 的首次出现行。

完整代码如下:

DeepAI

为天生具有创造力的人提供的AI工具

mask = df.ID.eq(1) & df.duplicated(subset=['ID'])
result = df[~mask].reset_index(drop=True)
print(result)

输出结果为:

   ID Value
0   1     A
1   2     B
2   3     A
3   2     C
4   3     A

✅ 注意事项:

  • duplicated() 默认 keep='first',即首次出现为 False,后续重复为 True,这正符合“保留首个 ID=1”的需求;
  • subset=['ID'] 明确限定去重依据列,避免误用其他列;
  • 若需对多个指定 ID(如 [1, 3])统一去重,可改用 df.ID.isin([1, 3]) 替代 eq(1);
  • 此方法不修改原 DataFrame,推荐用于链式操作或临时清洗场景。

该方案简洁、高效、可读性强,避免了 groupby 或循环等低效方式,是 Pandas 条件去重的典型范式。

相关推荐:

如何正确删除 XML 文件中的指定元素节点

使用Python的xml.etree.ElementTree删除XML元素时,直接通过索引delroot[i][j]无法可靠移除节点,因其不触发父节点的结构更新;应改用find()定位目标元素后调用parent.remove(child)方法。 使用python的`xml.etree.element...

Python中如何填充DataFrame中的空值_通过fillna函数指定填充策略

fillna默认不填充,必须显式传入value;支持字典分列填充、ffill/bfill前后向填充、interpolate数值插值,但需据空值语义谨慎选择策略。 fillna函数的基本用法和默认行为 fillna是pandas中最直接的空值填充方法,但它不是“智能补全”,而是按你指定的值或策略做静态...

如何在 Python 中提前终止程序执行以避免无效输入后的后续逻辑运行

本文讲解python中三种安全终止程序执行的方法:使用exit()、合理缩进控制流程、封装函数配合return,帮助初学者在用户输入非法值时立即停止代码运行,防止后续逻辑错误。 本文讲解python中三种安全终止程序执行的方法:使用exit()、合理缩进控制流程、封装函数配合return,帮助初学者...

如何在 Python 中提前终止程序执行以避免后续代码运行

当用户输入无效数字时,需立即停止程序并输出错误提示,防止继续执行图像打印和电脑选择逻辑。本文介绍三种主流解决方案:使用exit()退出、合理缩进控制流程,以及封装为函数后用return提前返回。 当用户输入无效数字时,需立即停止程序并输出错误提示,防止继续执行图像打印和电脑选择逻辑。本文介绍三种主流...

如何安全使用 write_pandas 避免 Snowflake 表中重复写入

本文解析AWSLambda环境下snowflake-connector-python[pandas]的write_pandas出现多份重复记录的根本原因——并非函数本身逻辑缺陷,而是Lambda异步调用机制与超时重试策略引发的并发重复执行,并提供可落地的排查路径与工程化防御方案。 本文解析awsla...