试图从Biopython获取分类信息

时间:2013-05-12 04:35:34

标签: list loops iteration bioinformatics biopython

我正在尝试更改以前使用biopython获取有关物种门信息的脚本。编写此脚本是为了一次检索一个物种的信息。我想修改脚本,以便我可以一次为100个生物做这个。 这是初始代码

import sys
from Bio import Entrez

def get_tax_id(species):
    """to get data from ncbi taxomomy, we need to have the taxid.  we can
    get that by passing the species name to esearch, which will return
    the tax id"""
    species = species.replace(" ", "+").strip()
    search = Entrez.esearch(term = species, db = "taxonomy", retmode = "xml")
    record = Entrez.read(search)
    return record['IdList'][0]

def get_tax_data(taxid):
    """once we have the taxid, we can fetch the record"""
    search = Entrez.efetch(id = taxid, db = "taxonomy", retmode = "xml")
    return Entrez.read(search)

Entrez.email = ""
if not Entrez.email:
    print "you must add your email address"
    sys.exit(2)
taxid = get_tax_id("Erodium carvifolium")
data = get_tax_data(taxid)
lineage = {d['Rank']:d['ScientificName'] for d in 
    data[0]['LineageEx'] if d['Rank'] in ['family', 'order']}

我已设法修改脚本,以便它接受包含我正在使用的生物之一的本地文件。但我需要将其扩展到100种生物体。 因此,我的想法是从我的生物体文件中生成一个列表,并以某种方式将列表中生成的每个项目分别输入到行taxid = get_tax_id("Erodium carvifolium")中,并用我的生物体名称替换“Erodium carvifolium”。但我不知道该怎么做。

以下是我的一些调整的代码示例版本

 import sys
from Bio import Entrez


def get_tax_id(species):
    """to get data from ncbi taxomomy, we need to have the taxid. we can
    get that by passing the species name to esearch, which will return
    the tax id"""
    species = species.replace(' ', "+").strip()
    search = Entrez.esearch(term = species, db = "taxonomy", retmode = "xml")
    record = Entrez.read(search)
    return record['IdList'][0]

def get_tax_data(taxid):
    """once we have the taxid, we can fetch the record"""
    search = Entrez.efetch(id = taxid, db = "taxonomy", retmode = "xml")
    return Entrez.read(search)

Entrez.email = ""
if not Entrez.email:
    print "you must add your email address"
    sys.exit(2)
list = ['Helicobacter pylori 26695', 'Thermotoga maritima MSB8', 'Deinococcus radiodurans R1', 'Treponema pallidum subsp. pallidum str. Nichols', 'Aquifex aeolicus VF5', 'Archaeoglobus fulgidus DSM 4304']
i = iter(list)
item = i.next()
for item in list:
     ???
taxid = get_tax_id(?)
data = get_tax_data(taxid)
lineage = {d['Rank']:d['ScientificName'] for d in
    data[0]['LineageEx'] if d['Rank'] in ['phylum']}
print lineage, taxid

问号是指我难以接受的地方。我不知道如何连接我的循环来取代?在get_tax_id(?)中。或者我是否需要以某种方式附加列表中的每个项目,以便每次修改它们以包含get_tax_id(Helicobacter pylori 26695),然后找到一些方法将它们放在包含taxid =

的行中

1 个答案:

答案 0 :(得分:2)

以下是您需要的内容,请将其放在您的功能定义下方,即在以下行中显示:sys.exit(2)

species_list = ['Helicobacter pylori 26695', 'Thermotoga maritima MSB8', 'Deinococcus radiodurans R1', 'Treponema pallidum subsp. pallidum str. Nichols', 'Aquifex aeolicus VF5', 'Archaeoglobus fulgidus DSM 4304']

taxid_list = [] # Initiate the lists to store the data to be parsed in
data_list = []
lineage_list = []

print('parsing taxonomic data...') # message declaring the parser has begun

for species in species_list:
    print ('\t'+species) # progress messages

    taxid = get_tax_id(species) # Apply your functions
    data = get_tax_data(taxid)
    lineage = {d['Rank']:d['ScientificName'] for d in data[0]['LineageEx'] if d['Rank'] in ['phylum']}

    taxid_list.append(taxid) # Append the data to lists already initiated
    data_list.append(data)
    lineage_list.append(lineage)

print('complete!')