因为您正在尝试解析系统发育树,我强烈建议让 BioPython 为您完成繁重的工作。
您可以使用Bio.Phylo 轻松解析和显示系统发育。然后它只是遍历所有树元素并在“at”符号处拆分名称。
因为 Phylo 期望输入在一个文件中,所以我们使用io.StringIO 创建一个内存中的类似文件的对象。获得完整的树就像
Phylo.read(io.StringIO(s), 'newick')
为了检查解析后的树是否正常,我使用print(tree) 打印一次。
现在我们要更改所有包含'@' 的节点名称。使用tree.find_elements,我们可以访问所有节点。有些节点没有名称,有些可能不包含'@'。所以要格外小心,我们首先检查if n.name and '@' in n.name。只有这样我们才能在'@' 处拆分每个节点的名称,并只取它的第一部分(索引 0):
n.name = n.name.split('@')[0]
为了重新创建初始字符串表示,我们使用Phylo.write:
out = io.StringIO()
Phylo.write(tree, out, "newick")
print(out.getvalue())
同样,write 想要获取文件参数 - 如果我们只想获取字符串,我们可以再次使用 StringIO 对象。
完整代码:
import io
from Bio import Phylo
if __name__ == '__main__':
s = '(Esy@ESY15_g64743_DN3_SP7_c0:0.0726396855636,Aar@AA_maker7399_1:0.137507902808,((Spa@Tp2g18720:0.0318934795022,Cpl@CP2_g48793_DN3_SP8_c:0.0273465005242):9.05326020871e-05,(((Bst@Bostr_13083s0053_1:0.0332592496158,((Aly@AL8G21130_t1:0.0328569260951,Ath@AT5G48370_1:0.0391706378372):0.0205924636564,(Chi@CARHR183840_1:0.0954469923893,Cru@Carubv10026342m:0.0570981548016):0.00998579652059):0.0150356382287):0.0340484449097,(((Hco@scaff1034_g23864_DN3_SP8_c_TE35_CDS100:0.00823215335663,Hlo@DN13684_c0_g1_i1_p1:0.0085462978729):0.0144626717872,Hla@DN22821_c0_g1_i1_p1:0.0225079453622):0.0206478928557,Hse@DN23412_c0_g1_i3_p1:0.048590776459):0.0372829371381):0.00859075940423,(Esa@Thhalv10004228m:0.0378509854703,Aal@Aa_G102140_t1:0.0712272454125):1.00000050003e-06):0.00328120860999):0.0129090235079):0.0129090235079;'
tree = Phylo.read(io.StringIO(s), 'newick')
print(' before '.center(20, '='))
print(tree)
for n in tree.find_elements():
if n.name and '@' in n.name:
n.name = n.name.split('@')[0]
print(' result '.center(20, '='))
out = io.StringIO()
Phylo.write(tree, out, "newick")
print(out.getvalue())
输出:
====== before ======
Tree(rooted=False, weight=1.0)
Clade(branch_length=0.0129090235079)
Clade(branch_length=0.0726396855636, name='Esy@ESY15_g64743_DN3_SP7_c0')
Clade(branch_length=0.137507902808, name='Aar@AA_maker7399_1')
Clade(branch_length=0.0129090235079)
Clade(branch_length=9.05326020871e-05)
Clade(branch_length=0.0318934795022, name='Spa@Tp2g18720')
Clade(branch_length=0.0273465005242, name='Cpl@CP2_g48793_DN3_SP8_c')
Clade(branch_length=0.00328120860999)
Clade(branch_length=0.00859075940423)
Clade(branch_length=0.0340484449097)
Clade(branch_length=0.0332592496158, name='Bst@Bostr_13083s0053_1')
Clade(branch_length=0.0150356382287)
Clade(branch_length=0.0205924636564)
Clade(branch_length=0.0328569260951, name='Aly@AL8G21130_t1')
Clade(branch_length=0.0391706378372, name='Ath@AT5G48370_1')
Clade(branch_length=0.00998579652059)
Clade(branch_length=0.0954469923893, name='Chi@CARHR183840_1')
Clade(branch_length=0.0570981548016, name='Cru@Carubv10026342m')
Clade(branch_length=0.0372829371381)
Clade(branch_length=0.0206478928557)
Clade(branch_length=0.0144626717872)
Clade(branch_length=0.00823215335663, name='Hco@scaff1034_g23864_DN3_SP8_c_TE35_CDS100')
Clade(branch_length=0.0085462978729, name='Hlo@DN13684_c0_g1_i1_p1')
Clade(branch_length=0.0225079453622, name='Hla@DN22821_c0_g1_i1_p1')
Clade(branch_length=0.048590776459, name='Hse@DN23412_c0_g1_i3_p1')
Clade(branch_length=1.00000050003e-06)
Clade(branch_length=0.0378509854703, name='Esa@Thhalv10004228m')
Clade(branch_length=0.0712272454125, name='Aal@Aa_G102140_t1')
==== result =====
(Esy:0.07264,Aar:0.13751,((Spa:0.03189,Cpl:0.02735):0.00009,(((Bst:0.03326,((Aly:0.03286,Ath:0.03917):0.02059,(Chi:0.09545,Cru:0.05710):0.00999):0.01504):0.03405,(((Hco:0.00823,Hlo:0.00855):0.01446,Hla:0.02251):0.02065,Hse:0.04859):0.03728):0.00859,(Esa:0.03785,Aal:0.07123):0.00000):0.00328):0.01291):0.01291;
Phylo 的默认格式使用的位数少于原始树中的位数。为了保持数字不变,只需用 '%s' 覆盖分支长度格式字符串:
Phylo.write(tree, out, "newick", format_branch_length="%s")