读取ATCG DNA序列,并计算第三位的ATCG数量

时间:2016-05-21 14:32:26

标签: python fasta

我想阅读ATCG DNA序列,并计算第三名的ATCG数量。

例如:

DNA = AAATTTCCCGGG

第三名ATCG是这样的:AA'A'TT'T'CC'C'GG'G'

所以在这个序列中A = 1 T = 1 C = 1 G = 1。

例如:

DNA = ATGGTATTTAAA

AT “G” GT “A” TT “T” AA “A”

我想计算3,6,9,12名ATCG号码。因此在DNA中A = 2 T = 1 C = 0 G = 1

我的txt文件是这样的:

>seq1
ATGGTATTTAAA
ATCGTTTTTAAA
>seq2
ATGGTATTTAAA
ATCGTTTTTAAA
ATCGTTTTTAAA
>seq3
ATGGTATTTAAA

我的代码是这样的:

f = open("a.txt","r")
seqlist = []
for line in f.readlines():
  line = line.strip("\n")
  if line.startswith(">"):
    print(line)
  elif line.startswith("A") or line.startswith("T") or line.startswith("C") or line.startswith("G"):
    seq = line
    y = 0
    for y in range(2, len(seq), 3):
      x = seq[y]
      print(x)

现在我可以获得第三名ATCG,我想把它放在一个列表中。

然后我可以算上ATCG。

但我不知道如何把它放在一个列表中。并获得以下结果。

seq1 A=3 T=3 C=1 G=1
seq2 A=? T=? C=? G=?
seq3 A=? T=? C=? G=?

非常感谢你的帮助。

1 个答案:

答案 0 :(得分:1)

这是一个尽可能少修改代码的选项:

from collections import Counter

counter = None
for line in f.readlines():
    line = line.strip("\n")
    if line.startswith(">"):
        if counter is not None:
            print(counter)
        print(line)
        counter = Counter()
    elif line.startswith("A") or line.startswith("T") or line.startswith("C") or line.startswith("G"):
        seq = line
        y = 0
        for y in range(2, len(seq), 3):
            x = seq[y]
            counter[x] += 1
print(counter)

输出:

>seq1
Counter({'A': 3, 'T': 3, 'C': 1, 'G': 1})
>seq2
Counter({'T': 5, 'A': 4, 'C': 2, 'G': 1})
>seq3
Counter({'A': 2, 'T': 1, 'G': 1})

除了改进整体代码以及更好地格式化输出之外,还有同样的事情:

from collections import Counter

counter = None
bases = 'ATCG'

def print_counter():
    print(' '.join('%s=%s' % (k, counter[k]) for k in bases))

with open("a.txt", "r") as f:  # Always open files like this
    for line in f:  # no need for readlines
        line = line.strip("\n")
        if line.startswith(">"):
            if counter is not None:
                print_counter()
            print(line)
            counter = Counter()
        elif line and line[0] in bases:
            counter.update(line[2::3])
print_counter()

输出:

>seq1
A=3 T=3 C=1 G=1
>seq2
A=4 T=5 C=2 G=1
>seq3
A=2 T=1 C=0 G=1