新人求助 python过滤算法

t464961268

test1.txt文件内容格式如图：



test2.txt文件格式如图：



想用过滤算法实现：test2的第一列中，只要出现与test1不同的号码，就把test2中这个号码以及后面的相关信息全删掉。
比如：
test2.txt中的5093556233在test1中没有找到，
则删除这十个5093556233，以及后面的1 4 1；1 9 1；0 7 5；~；6 36 2这十组数。


初学python，求大神指点

mswsg

因为你的数据量未知，随手写了一个，不知道速度怎样。
思路是这样：首先将a =  set(test1.txt) 文件放入集合，
逐行读取test2.txt文件，判断 line[0] in a，是就输出

1. # encoding: utf-8
   
2. 
   
3. with open('test1.txt', 'r')as f1, open('test2.txt', 'r') as f2:
   
4.     a = set(line.strip()[:-1] for line in f1)
   
5.     print a
   
6.     for line in f2:
   
7.         line = line.strip().split()
   
8.         if line[0] in a:
   
9.             print ' '.join(line)

复制代码

t464961268

回复 2# mswsg


    test2文件有5万多个数据。。test1有1570.。非常感谢。

t464961268

回复 2# mswsg


    非常感谢您的指点，已经获得我所需要的数据了，非常感谢！