From 765a73ed7bb055d11aea2627af8daa9cfc6d664b Mon Sep 17 00:00:00 2001 From: Calvin Morrison Date: Wed, 16 Jul 2014 14:06:21 -0400 Subject: add working remove mers from file --- src/remove_mers_from_file.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/remove_mers_from_file.py b/src/remove_mers_from_file.py index bef3ea3..3576e35 100755 --- a/src/remove_mers_from_file.py +++ b/src/remove_mers_from_file.py @@ -8,9 +8,12 @@ if __name__ == "__main__": mers_to_delete = set() - for mer in open(sys.argv[1], 'r'): - mer = mer.strip().split()[0] - mers_to_delete.add(mer) + for mer in open(sys.argv[1], 'rU'): + sp = mer.strip().split() + if len(sp) == 2: + mer = sp[0] + mers_to_delete.add(mer) + for line in sys.stdin: if line.split()[0] not in mers_to_delete: -- cgit v1.2.3