树结构的打印——Python实现

最新推荐文章于 2024-06-13 09:40:06 发布

frostime

最新推荐文章于 2024-06-13 09:40:06 发布

阅读量2.9k

点赞数 1

分类专栏：数据结构&算法 Python 文章标签： python 树结构

本文链接：https://blog.csdn.net/frostime/article/details/104809060

版权

数据结构&算法同时被 2 个专栏收录

13 篇文章 1 订阅

订阅专栏

Python

12 篇文章 3 订阅

订阅专栏

在做 HTML 的解析的时候遇到的需求，希望能用 Python 实现一下类似 bash 下 tree 命令输出的效果，类似这样:

.
├── a3c_demo.py
├── dqn-boat_mannual
│   ├── ICONS_Python
│   ├── Image
│   ├── __pycache__
│   ├── ddpg.py
│   ├── enviroment.py
│   ├── gameui.py
│   ├── ima1.py
│   ├── main.py
│   ├── manual_control.py
│   └── temp.py
├── dqn-boat_mannual.7z
├── envdemo.py
└── log
    └── events.out.tfevents.1582014010.ZYPLAPTOP

想了一下发现，也不难实现，主要就是用 DFS 递归打印，注意判断尾部节点就行了。示例代码放到这里，说不定以后用的上:

# -*- coding: utf-8 -*-
import json
import re
from queue import Queue
from pprint import pprint

from lxml import etree


class Node:

    def __init__(self, text, level):
        self.text = text
        self.level = level
        self.children = []

    def addChild(self, node):
        self.children.append(node)

    def empty(self):
        return len(self.children) == 0

    def __str__(self):
        return self.text


def printTree(node, indent: list, final_node=True):
    """打印树结构的算法
        Parameters:
        ----------
            node: 节点
            indent: 记录了节点之前需要打印的信息
            final_node: node 是否是最后一个节点（i.e. 没有下一个 sibling 了）
    """
    for i in range(node.level):
        print(indent[i], end='')
    if final_node:
        print('└──', end='')
    else:
        print('├──', end='')
    print(node.text)

    if node.empty():
        return
    else:
        cnt = len(node.children)
        for i, n in enumerate(node.children):
            c = '    ' if final_node else '│   '
            indent.append(c)
            last_node = i == cnt - 1
            printTree(n, indent, last_node)
            del indent[-1]



def inspectElem(elem: etree.Element, level=0):
    """递归地建立 HTML 树"""
    tag = str(elem.tag)
    node = Node(tag, level)
    for child in elem.iterchildren():
        sub_node = inspectElem(child, level + 1)
        node.addChild(sub_node)
    return node

if __name__ == "__main__":
    f_path = r"./sample.html"
    with open(f_path, encoding='utf-8') as f:
        html = f.read()
    tree = etree.HTML(html)
    node = inspectElem(tree)
    indent = []
    print('.')
    printTree(node, indent, True)

这里的案例是打印 HTML 标签的嵌套结构，源文件内容如下:

<!DOCTYPE html>
<html lang="zh-cn">

<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <meta http-equiv="X-UA-Compatible" content="ie=edge">
    <title>Table布局</title>
</head>
<!--去除边缘，否则灰色区域旁边会有白边-->

<body marginheight="0px" marginwidth="0px">
    <table width="100%" height="950px" style="background-color:darkgray;">
        <tr>
            <td colspan="2" width="100%" height="10%" style="background-color:cadetblue;">
                这是头部
            </td>
        </tr>
        <tr>
            <td width="30%" height="80%" style="background-color:darkkhaki;">
                左菜单
            </td>
            <td width="70%" height="80%" style="background-color:cornflowerblue;">
                右菜单
            </td>
        </tr>
        <tr>
            <td colspan="2" width="100%" height="10%" style="background-color: goldenrod;">
                底部
            </td>
        </tr>
    </table>
</body>

</html>

打印效果如下:

.
└──html
    ├──head     
    │   ├──meta 
    │   ├──meta 
    │   ├──meta 
    │   └──title
    ├──<cyfunction Comment at 0x00000258F44AE288>
    └──body
        └──table
            ├──tr
            │   └──td
            ├──tr
            │   ├──td
            │   └──td
            └──tr
                └──td