You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何基于含重复元素的列表提取数据集中未匹配次数的行?

解决思路与实现代码

我明白你的问题了——你要的不是完全排除packed里出现过的Pid,而是只保留那些超出packed中对应Pid出现次数的行,对吧?之前用isin的思路不对,因为它只判断Pid是否存在于列表中,完全忽略了重复次数的差异。

核心思路

  1. 先统计packed列表中每个Pid的出现次数
  2. 对数据集按Pid分组,给每组内的行按顺序编号(从0开始)
  3. 筛选出编号大于等于packed中对应Pid次数的行——这些就是超出列表覆盖次数的额外行

具体实现代码

方法一:使用collections.Counter统计次数

from collections import Counter
import pandas as pd

# 你的原始列表
packed = [1, 5, 8, 2, 3, 3, 7, 3, 7, 7, 4, 6, 3]
# 统计每个Pid在packed中的出现次数
pid_packed_counts = Counter(packed)

# 假设你的数据集是data,包含"Pid"列
# 给每个Pid组内的行添加顺序编号(从0开始)
data['row_index_in_group'] = data.groupby('Pid').cumcount()

# 筛选出超出packed次数的行:组内编号 >= 该Pid在packed中的次数
new_df = data[
    data['row_index_in_group'] >= data['Pid'].map(pid_packed_counts).fillna(0)
].reset_index(drop=True)

# 可以删掉临时添加的编号列
new_df = new_df.drop('row_index_in_group', axis=1)

方法二:用Pandas自带的value_counts统计次数

如果你不想引入collections模块,也可以用Pandas来统计:

import pandas as pd

packed = [1, 5, 8, 2, 3, 3, 7, 3, 7, 7, 4, 6, 3]
pid_packed_counts = pd.Series(packed).value_counts()

data['row_index_in_group'] = data.groupby('Pid').cumcount()
new_df = data[
    data['row_index_in_group'] >= data['Pid'].map(pid_packed_counts).fillna(0)
].reset_index(drop=True)

new_df = new_df.drop('row_index_in_group', axis=1)

为什么原来的代码不生效?

你之前写的new_df= data[~data["Pid"].isin(packed)].reset_index(drop=True),逻辑是排除所有Pid在packed里的行,但你实际需要的是保留那些Pid在packed里,但出现次数超过packed中数量的行——这两个需求完全不同,所以原代码无法得到你想要的结果。

内容的提问来源于stack exchange,提问作者Rahul Sharma

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.13 06:37:49