階層構造テキストを可視化する方法を探しています

Question

前提・実現したいこと

【前提】
・階層構造のテキストファイルがあります（以下の例参照）
・例（A～Gのテキスト。「.」が階層レベルを示す。ファイル名「test.txt」とする。テキストは100個ほど存在。）
.A
..B
.C
..D
..E
...F
.G

【実現したいこと】
・このテキストを以下のような階層構造としてpythonを用いて可視化したいです。
・最終出力はテキストでなくてもかまいません。（png等でもOK）
・以下のような垂れ下がり型（？）の形状を希望（扇型に広がったいるのはNG）

test
│
├─ A
│ │
│ └─ B
│
├─ C
│ │
│ ├─ D
│ │
│ └─ E
│ │
│ └─ F
│
└─ G

発生している問題・エラーメッセージ

・扇形に書けるものはgraphvizなど色々見つかったのですが、
垂れ下がり型で書ける方法が分かりません。
・唯一以下のURLに垂れ下がり型でテキスト出力の物があったのですが、
元テキスト(test.txt)から階層構造を自動出力する方法が構築できず途方に暮れています。
参考URL：https://rosettacode.org/wiki/Visualize_a_tree#Python

該当のソースコード（参考URLより拝借）

'''Visualize a tree'''
 
from itertools import (chain, repeat, starmap)
from operator import (add)
 
 
'''drawTree :: Tree a -> String'''
def drawTree(tree):
    '''ASCII diagram of a tree.'''
    return '\n'.join(draw(tree))
 
 
'''draw :: Tree a -> [String]'''
def draw(node):
    '''List of the lines of an ASCII
       diagram of a tree.'''
    def shift(first, other, xs):
        return list(starmap(
            add,
            zip(
                chain([first], repeat(other, len(xs) - 1)),
                xs
            )
        ))
 
    def drawSubTrees(xs):
        return (
            (
                ['│'] + shift(
                    '├─ ', '│  ', draw(xs[0])
                ) + drawSubTrees(xs[1:])
            ) if 1 < len(xs) else ['│'] + shift(
                '└─ ', '   ', draw(xs[0])
            )
        ) if xs else []
 
    return (str(root(node))).splitlines() + (
        drawSubTrees(nest(node))
    )
 
 
''' TEST ----------------------------------------------------'''
'''main :: IO ()'''
def main():
    '''Test'''
 
    '''tree :: Tree Int'''
    tree = Node(1)([
        Node(2)([
            Node(4)([
                Node(7)([])
            ]),
            Node(5)([])
        ]),
        Node(3)([
            Node(6)([
                Node(8)([]),
                Node(9)([])
            ])
        ])
    ])
 
    print(drawTree(tree))
 
 
''' GENERIC -------------------------------------------------'''
 
 
''' Node :: a -> [Tree a] -> Tree a'''
def Node(v):
    '''Contructor for a Tree node which connects a
       value of some kind to a list of zero or
       more child trees.'''
    return lambda xs: {'root': v, 'nest': xs}
 
 
''' nest :: Tree a -> [Tree a]'''
def nest(tree):
    '''Accessor function for children of tree node.'''
    return tree['nest'] if 'nest' in tree else None
 
 
''' root :: Dict -> a'''
def root(dct):
    '''Accessor function for data of tree node.'''
    return dct['root'] if 'root' in dct else None
 
 
''' MAIN ---'''
if __name__ == '__main__':
    main()```

試したこと

上記のコードを用いて手作業でNodeを構築すれば、階層構造を作成できたのですが、
元テキストから自動で階層構造を構築する方法が全く分かりません。

補足情報（FW/ツールのバージョンなど）

python3.7
anaconda 環境

Comments

[YND_teratail]

修正のお願いがありますが、https://apuridasuo.hatenablog.com/entry/2019/10/20/220308#%E3%82%BD%E3%83%BC%E3%82%B9%E3%82%B3%E3%83%BC%E3%83%89%E3%82%92%E8%B2%BC%E3%82%8B%E3%81%A8%E3%81%8D%E3%81%AF%E3%82%BF%E3%82%B0%E3%81%A7%E5%9B%B2%E3%82%80のように、<code>ボタンを押して、ソースコードを見やすくしてもらえないでしょうか。

[hhiro45fd]

すいません、失念してました。修正いたしました。

[YND_teratail]

ありがとうございます。

Answer 1

ライブラリ anytree を使った方法

import fileinput
from anytree import Node, RenderTree # 要: pip install anytree

# ルートノードのみ準備
root = Node("test")
stack = [root]

for line in fileinput.input(): # 標準入力 or 引数のファイルから読込

    # 入力データのparse
    # ※ ここは実際のデータに合わせて要調整 (depth: 階層の深さ、value: 表示する値)
    # 例: "...C\n" -> depth:3, value:"C"
    depth = line.count(".") # ドットの数をノードの深さ
    value = line.lstrip(".").rstrip() # ドットを取り除いた部分を値
    assert "." not in value

    # 木構造データの構築
    while len(stack) > depth:
        stack.pop()
    node = Node(value, parent=stack[-1])
    stack.append(node)

# テキスト出力: 以下は anytree ライブラリのサンプルコードそのまま
for pre, fill, node in RenderTree(root):
    print("%s%s" % (pre, node.name))

---

importer/exporter で複数のデータの読み込みと書き出しの方法を選択できます。

anytree ライブラリのブランチに indented importer というものがあったそうです。
最近の情報を見ると 3日前に PR が投稿されてました。
https://github.com/c0fec0de/anytree/pull/155

過去のバージョンで使えていて、保守から外れた経緯があるようなので、
PRの実装が採用されるかどうかは解りませんが、
コード自体はドットをスペースに変換すればそのまま使えそう。

---

質問に掲載のコードで改善可能な点

ソースコード中のコメント部分から Haskell っぽい所があったので言及しておくと

drawTree関数の実装
print("\n".join(...)) では、全ての出力結果のリストを作ってから
改行文字で連結し、一度に出力という処理順序になる為、
データ数が多いと、最初に何も表示されないまま少し時間が掛かかります。

PythonでHaskellのような遅延評価をするには、for文を使い、リストの扱いを
イテレーターやジェネレーターにすることで、一行ずつ出力が可能です。

Comments

[hhiro45fd]

ご回答ありがとうございます。

非常に詳細にありがとうございます。あっさりとできてしまいました。
anytree・・・不勉強なため初耳でした。
補足の部分も併せてじっくり読ませて頂きます。

Answer 2

ちゃんと表示されるかわかりませんが、少なくともサンプルの場合はできます

import re
pRt="├─"
pLRt="└─"
pTt="│"
pSp="  "
filename="File1"
file = open(filename, "r")
lines = file.readlines()
file.close()
nextDotNum=0
tree=""
nextRt=100
for line in reversed(lines):
    dotNum=len(re.search(r"^.+",line).group())
    str=re.search(r"(?<=.)[^.]+$",line).group().strip()
    brc=""
    if(dotNum>1):
        brc=pTt
        for i in range(dotNum-1):
            brc+=pSp
    if(dotNum==nextDotNum):
        pR=pRt
    else:
        print(str)
        if(dotNum>nextDotNum):
            pR=pLRt
        elif(dotNum==nextDotNum):
            pR=pRt
        elif(dotNum>nextRt):
            pR=pLRt
        else:
            pR=pRt
    brc="\n"+brc+pTt+"\n"+brc+pR+" "+str
    tree=brc+tree
    nextDotNum=dotNum
    if(dotNum>nextDotNum):
        nextRt=nextDotNum
    if(dotNum<nextRt):
        nextRt=dotNum
tree=filename+tree
print(tree)

Comments

[hhiro45fd]

ご回答ありがとうございます。

いくつかテストしてみましたが、今の所問題はなさそうです。
考え方が非常に参考になりました。ありがとうございます。

Answer 3

半分冗談ですが頑張って作りました。

$ cat hoge.py
#!/usr/bin/env python3
buf="""
.A
..B
.C
..D
..E
...F
.G
"""

parent=['.','.','.','.','.']
for line in buf.split("\n"):
    i=0
    for c in list(line):
        if c == '.':
            print(parent[i]+"/", end='')
        else:
            print(c+"/", end='')
            parent[i]=c
        i+=1
    print("")

$ ./hoge.py

./A/
./A/B/
./C/
./C/D/
./C/E/
./C/E/F/
./G/

$ mkdir -p `./hoge.py`
$ tree
.
├── A
│   └── B
├── C
│   ├── D
│   └── E
│       └── F
└── G

7 directories, 3 files

Comments

[68user]

標準入力から読めるんですね…。
$ (echo ほげ; echo ほげ/ふが; echo あいう; echo ほげ/ううう) |tree --fromfile=.
.
├── あいう
└── ほげ
　　├── ううう
　　└── ふが

[hhiro45fd]

ご回答ありがとうございます。

なんという荒技！
真面目にやったら名称に結構制約が厳しいようで残念ながら諦めました・・・
（大変面白かったです、ありがとうございます<(_ _)>）