astrofrog / astrodendro-deprecated Goto Github PK

from astrodendro import Dendrogram
d = Dendrogram(ppv_data)
trunk = d.trunk[0]


def walk(trunk, nodes=[]):
    import astrodendro.components as comp
    for item in trunk.items:
        if type(item) is comp.Branch:
            print(item.items)
            nodes.append(len(item.items))
            walk(item)
        else:
            pass
    return nodes

res = walk(trunk)

print(len(res))