python - ETE2 : adding image to nodes

Question

我一直在尝试使用 Python 中的 ete2 软件向系统发育树中的节点添加不同的图像，但没有成功。

from ete2 import Tree, TreeStyle, NodeStyle, TextFace, faces, add_face_to_node, AttrFace
ts.show_leaf_name = True
ts.show_branch_support = True

nw = """
(((Dre:0.008339,Dme:0.300613)1.000000:0.596401,
(Cfa:0.640858,Hsa:0.753230)1.000000:0.182035)1.000000:0.106234,
((Dre:0.271621,Cfa:0.046042)1.000000:0.953250,
(Hsa:0.061813,Mms:0.110769)1.000000:0.204419)1.000000:0.973467);
"""
t = Tree(nw)


img_path = "/home/leonard/Desktop/img_faces/"

humanFace = faces.ImgFace(img_path+"human.png")

mouseFace = faces.ImgFace(img_path+"mouse.png")


def my_layout(node):

 if name.startswith("Dre"):

  faces.add_face_to_node(humanface, node, column=1)

t.show(my_layout)

ts = TreeStyle()

t.render("img_faces.png", w=600, tree_style = ts)

这些是我收到的错误消息:

 File "abc1.py", line 34, in <module>
    t.show(my_layout)
  File "/usr/local/lib/python2.7/dist-packages/ete2-2.1rev544-py2.7.egg/ete2/coretype/tree.py", line 1283, in show
    drawer.show_tree(self, layout=layout, tree_style=tree_style)
  File "/usr/local/lib/python2.7/dist-packages/ete2-2.1rev544-py2.7.egg/ete2/treeview/drawer.py", line 84, in show_tree
    tree_item, n2i, n2f = render(t, img)
  File "/usr/local/lib/python2.7/dist-packages/ete2-2.1rev544-py2.7.egg/ete2/treeview/qt4_render.py", line 258, in render
    set_style(n, layout_fn)
  File "/usr/local/lib/python2.7/dist-packages/ete2-2.1rev544-py2.7.egg/ete2/treeview/qt4_render.py", line 746, in set_style
    layout_func(n)
  File "abc1.py", line 29, in my_layout
    if name.startswith("Dre"):
NameError: global name 'name' is not defined

非常感谢任何帮助!

Answer 1

这是来自 google group 的 Jaime Huerta Cepas 的解决方案，我引用:

“您的脚本中存在一般性 Python 编程错误(“name”变量不存在)。我猜您的意思是“node.name.startswith()”而不是“name.startswith()””

成功了。