I have a fasta file that I am parsing. The file is composed of several sequences that belongs to the same gene from a different bacterial strain. What I want to do, and the script does is to check if the sequence are equal or different to the reference seq. With that information I want to produce a new file BUT I only one one sequence of each.
def checking_sequences():
gene_records=list(SeqIO.parse('/files/gene_A.fasta', 'fasta'))
ref_id=gene_records[-1].id
ref_seq=gene_records[-1].seq
#print gene_records[-1].description
output_handle=open('//files/' + 'corrected_gene_1', 'a')
print len(gene_records)
count=0
dif_count=0
reference_list=[]
for gene_record in gene_records:
#count+=1
if len(gene_record.seq) == len(ref_seq):
#print len(gene_records.seq)
#print len(ref_seq)
print 'Found all lengths are equal'
else:
print 'Found %s sequence with different lengths' % (gene_records.description)
###checking sequence equality
if gene_record.seq==ref_seq:
count+=1
gene_record.id=gene_record.id +'_0'
reference_list.append(gene_record)
ref_count=reference_list.count(gene_record.seq)
print 'There are %i sequences are equal to the reference sequence' %(count)
else:
dif_count+=1
reference_list.append(gene_record.seq)
seq_count=reference_list.count(gene_record.seq)
gene_record.id=gene_record.id +'_'+ str(dif_count)
print 'Found %i different that ref_seq' % (seq_count)
print 'xxxxxxxxxxx'
#print seq_count
#print len(reference_list)
SeqIO.write(gene_record, output_handle, 'fasta')
output_handle.close()
checking_sequences() For some clarification :
original file desire output
>gene_1 strainIDx >gen1_strainIDx
seqA seqA
>gene_1 strainIDy >gene_1 strainIDy
seqB seqB
>gene_1 strainIDz
seqA
I don't mind about the ID just I would like to have one seq of each. I have tried to use 'break'but I don't get the output I would like to. Help will be appreciate