python DataFrame 取差集实例

时间:2021-05-22

需求:给定一个dataframe和一个list,list中存放的是dataframe中某一列的元素,删除dataframe中与list元素重复的行(即取差集)。

在网上搜了一圈,好像没看到DataFrame中取差集的方式,所以自己写了一个。方法比较繁琐,如果有更简便的方式,请留言。

import pandas as pddata = [[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]# 重置indexres.index = [i for i in range(len(res))]print(res)

以上这篇python DataFrame 取差集实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章