如何基于含重复元素的列表提取数据集中未匹配次数的行?
解决思路与实现代码
我明白你的问题了——你要的不是完全排除packed里出现过的Pid,而是只保留那些超出packed中对应Pid出现次数的行,对吧?之前用isin的思路不对,因为它只判断Pid是否存在于列表中,完全忽略了重复次数的差异。
核心思路
- 先统计
packed列表中每个Pid的出现次数 - 对数据集按Pid分组,给每组内的行按顺序编号(从0开始)
- 筛选出编号大于等于
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
相关产品推荐
相关产品推荐

