python DataFrame 取差集

最新推荐文章于 2025-05-25 16:32:02 发布

hiptonese

最新推荐文章于 2025-05-25 16:32:02 发布

阅读量2.2w

点赞数 3

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.youkuaiyun.com/u012560212/article/details/78626273

本文介绍了一种在Pandas DataFrame中实现差集操作的方法，即如何从一个DataFrame中删除另一列表中存在的行。通过示例代码展示了如何找到并移除指定列中与列表元素匹配的行，最终获取差集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

需求：给定一个dataframe和一个list，list中存放的是dataframe中某一列的元素，删除dataframe中与list元素重复的行（即取差集）。
在网上搜了一圈，好像没看到DataFrame中取差集的方式，所以自己写了一个。方法比较繁琐，如果有更简便的方式，请留言。

import pandas as pd
data = [[1,2,3],[2,3,4],[3,4,5],[4,5,6]]
# 创建dataframe，包含a,b,c三列
df = pd.DataFrame(data, columns=['a','b','c'])
print(df)
# a 列中待删除的元素
a_to_drop = [1, 3]
# 找到待删除元素所在的位置，返回的是 true or false 序列
flag = df['a'].isin(a_to_drop)
# 由于我们要取差集，因此对上述序列取反
diff_flag = [not f for f in flag]
# res 为我们所需要的差集
res = df[diff_flag]
# 重置index
res.index = [i for i in range(len(res))]
print(res)