pandas按若干个列的组合条件筛选数据的方法

时间:2021-05-22

还是用图说话

A文件:

比如,我想筛选出“设计井别”、“投产井别”、“目前井别”三列数据都为11的数据,结果如下:

当然,这里的筛选条件可以根据用户需要自由调整,代码如下:

# -*- coding: utf-8 -*-"""Created on Wed Nov 29 10:46:31 2017@author: wq"""import pandas as pd#input.csv是那个大文件,有很多很多行df1 = pd.read_csv(u'input.csv', encoding='gbk')#加encoding=‘gbk'是因为文件中存在中文,不加可能出现乱码#这里的筛选条件可以根据用户需要进行修改outfile = df1[(df1[u'设计井别']=='11') & (df1[u'投产井别']=='11') &(df1[u'目前井别']=='11')]outfile.to_csv('outfile.csv', index=False, encoding='gbk')

有时我们也会有相反的一个需求,需要删除“设计井别”、“投产井别”、“目前井别”三列数据都为11的那些行,效果如下:

代码如下:

#input.csv是那个大文件,有很多很多行df1 = pd.read_csv(u'input.csv', encoding='gbk')df2 = pd.read_csv(u'outfile.csv', encoding='gbk')#加encoding=‘gbk'是因为文件中存在中文,不加可能出现乱码index = ~df1[u'汉字井号'].isin(df2[u'汉字井号'])df4 = df1[index]df4.to_csv('outfile1.csv', index=False, encoding='gbk')

以上这篇pandas按若干个列的组合条件筛选数据的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

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

相关文章