[Python学习]ete3绘制树形图的源代码Excel数据可视化.pdfVIP

【Python学习】ete3绘制树形图的源代码，Excel数

据可视化

用python的ete3绘制树形图的源代码文件：ete3_test99.py，

运行可生成指定数据（如A0000002）的所有下级树的树形结构图。

源代码用pycharm编辑。代码如下：

importete3

importpandasaspd

fromete3importTree,NodeStyle,TreeStyle

#fromete3.webtoolimportWebpage

df=pd.read_excel(DATA.xlsx)

boss=A0000002

subordinates=df[df[上级ID]==boss][下级ID].tolist()

defadd_subtree(subtree,t):

forchild_nodeinsubtree:

ts=t.add_child(name=child_node)

children=df[df[上级ID]==child_node][下级

ID].tolist()

ifchildren:

add_subtree(children,ts)

#节点标签、样式

label_size=10

name_face=ete3.TextFace(child_node,

fgcolor==,fsize=label_size)

ts.add_face(name_face,column=0,position=branch-

right)

ns=NodeStyle()

ns[fgcolor]=green

ns[size]=20

ns[shape]=circle#节点形状，emptyCircle使用

空心圆circle实心圆

#ns.show_leaf_name=True

ts.set_style(ns)

defdraw_tree(boss,Webpage=None):

t=Tree(name=boss)

add_subtree(subordinates,t)

#主节点标签、样式

name_face_1=ete3.TextFace(boss,fgcolor==,

fsize=15)

t.add_face(name_face_1,column=0,position=branch-

right)

ns=NodeStyle()

ns[bgcolor]==#背景颜色