Question

我在Rosalind上遇到了一个问题，该问题涉及输出具有最高GC含量百分比和该百分比的字符串的ID，而我遇到了问题。我能够输出样本数据ID +百分比，但我不知道如何找到最高的。

#!/bin/env python3

import sys

input = input()
file = open(input, "r")
#print(input)

gc = 0
at = 0
unknown = 0

for line in file:
  if line.startswith(">"):
    if (gc + at) > 0:
     total = gc + at
     percentage = float(gc)/float(total) * 100
     result = percentage
     print(seq_id, result)
    seq_id = line.strip()
    gc = at = unknown = 0
  else:
   nuc_str = list(line.strip())
   for n in nuc_str:
    if n == "G" or n == "g" or n == "C" or n == "c":
     gc += 1.0
    elif n == "A" or n == "a" or n == "T" or n == "t":
     at += 1.0
    else:
     unknown += 1.0
total = gc + at
percentage = float(gc)/float(total) * 100
result = percentage
print(seq_id, result)

所需的输出是：

Rosalind_0808
60.919540

我得到的输出是：

>Rosalind_6404 53.75
>Rosalind_5959 53.57142857142857
>Rosalind_0808 60.91954022988506

输入文件是取自rosalind的样本数据fasta文件：

>Rosalind_6404
CCTGCGGAAGATCGGCACTAGAATAGCCAGAACCGTTTCTCTGAGGCTTCCGGCCTTCCCTCCCACTAATAATTCTGAGG
>Rosalind_5959
CCATCGGTAGCGCATCCTTAGTCCAATTAAGTCCCTATCCAGGCGCTCCGCCGAAGGTCTATATCCATTTGTCAGCAGACACGC
>Rosalind_0808
CCACCCTCGTGGTATGGCTAGGCATTCAGGAACCGGAGAACGCTTCAGACCAGCCCGGACTGGGAACCTGCGGGCAGTAGGTGGAAT

我正在尝试不使用biopython。

Answer 1

可以使用一些优化来使您的代码更好，更简洁。例如，使用list中的tuples，可以使序列标头与其GC内容相关联。

所以您的循环看起来像：

# Create an empty list to collect results
results = []
for line in file:
    if line.startswith(">"):
        if (gc + at) > 0:
            total = gc + at
            percentage = float(gc)/float(total) * 100
            result = percentage
            # Add results to the list here
            results.append((seq_id, result))
            seq_id = line.strip()
            gc = at = unknown = 0
    else:
        nuc_str = list(line.strip())
        for n in nuc_str:
            if n == "G" or n == "g" or n == "C" or n == "c":
                gc += 1.0
            elif n == "A" or n == "a" or n == "T" or n == "t":
                at += 1.0
            else:
                unknown += 1.0

运行循环后，results如下：

[
 ("Rosalind_6404", 53.75), 
 ("Rosalind_5959", 53.57142857142857), 
 ("Rosalind_0808", 60.91954022988506),
]

然后，要收集具有最高GC含量的序列，我们可以使用itemgetter：

from operator import itemgetter
result = max(results, key=itemgetter(1))

我们使用key参数告诉max在元组中找到位置1处具有最高值的列表项。
result看起来像这样：

("Rosalind_0808", 60.91954022988506)

要以所需格式输出结果，我们可以使用format：

 print("{}\n{:.6f}".format(*result))

哪个输出：

Rosalind_0808
60.919540

其他优化：

使用with语句打开文件，这样做的好处是在处理后可以正确关闭文件
计算GC，并使用序列长度（来自len）作为除数
使用lower()而不是比较大写和小写

我写了自己的版本，看起来像这样：

from operator import itemgetter

res = []
with open("test.fa", "r") as fh:
    for line in fh:
        line = line.strip()
        if line[0] in ">":
            header = line[1:]
            seq = next(fh).strip()
            gc = sum(1 for base in seq if base.lower() in "gc")
            res.append((header, gc / len(seq) * 100))

result = max(res, key=itemgetter(1))
print("{}\n{:.6f}".format(*result))

如何找到包含ID最高的GC内容百分比？

1 个答案: