diff options
Diffstat (limited to 'src/remove_mers_from_file.py')
| -rwxr-xr-x | src/remove_mers_from_file.py | 9 |
1 files changed, 6 insertions, 3 deletions
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: |