3

要約すれば

SO question: Python file parsing: Build tree from text fileの関数を使用して、テキスト ファイル入力からツリー構造を作成しています。しかし、グローバル変数を使用してのみツリーを作成でき、これを回避する方法が見つかりません。

入力データ

というファイルdata.txtには、次のものがあります。

Root
-A 10
-B
--B A 2
--B B 5
--B Z 9
--B X
---X 4
---Y
----Y0 67
----Y1 32
---Z 3
-C 19

望ましい結果

{'B': ['B A 2', 'B B 5', 'B Z 9', 'B X'],
 'B X': ['X 4', 'Y', 'Z 3'],
 'Root': ['A 10', 'B', 'C 19'],
 'Y': ['Y0 67', 'Y1 32']}

マイコード

import re, pprint
PATTERN = re.compile('^[-]+')
tree = {}

def _recurse_tree(parent, depth, source):
    last_line = source.readline().rstrip()
    while last_line:
        if last_line.startswith('-'):
            tabs = len( re.match(PATTERN, last_line).group() )
        else:
            tabs = 0
        if tabs < depth:
            break
        node = re.sub(PATTERN, '', last_line.strip())
        if tabs >= depth:
            if parent is not None:
                print "%s: %s" %(parent, node)
                if parent in tree:
                    tree[parent].append(node)
                else:
                    tree[parent] = [ node, ]
            last_line = _recurse_tree(node, tabs+1, source)
    return last_line

def main():
    inFile = open("data.txt")
    _recurse_tree(None, 0, inFile)
    pprint.pprint(tree)

if __name__ == "__main__":
    main()

問題

グローバル変数を取り除くにはどうすればよいですtreeか? 私が行うすべてのことは、コードをはるかに長くまたは醜くするように見えますが、関数を頻繁に使用したいので、コア結果の副作用に依存するのは嫌いです。

補足

以下の回答の後、次の方法で を返すようにコードを修正しtreeました。これはpythonicですか？タプルを返してから最初の要素を捨てるのは、洗練されていないように思えます。

def _recurse_tree(parent, depth, source, tree=None):
    if tree is None:
        tree = {}
    last_line = source.readline().rstrip()
    while last_line:
        if last_line.startswith('-'):
            tabs = len( re.match(PATTERN, last_line).group() )
        else:
            tabs = 0
        if tabs < depth:
            break
        node = re.sub(PATTERN, '', last_line.strip())
        if tabs >= depth:
            if parent is not None:
                print "%s: %s" %(parent, node)
                if parent in tree:
                    tree[parent].append(node)
                else:
                    tree[parent] = [ node, ]
            last_line, tree = _recurse_tree(node, tabs+1, source, tree)
    return last_line, tree

def main():
    inFile = open("data.txt")
    tmp, tree = _recurse_tree(None, 0, inFile)
    pprint.pprint(tree)
4

3 に答える 3

4

あなたのtree変数はすでに変更可能です。再帰呼び出しと一緒に渡すだけです：

def _recurse_tree(parent, depth, source, tree=None):
    if tree is None:
        tree = {}

    last_line = source.readline().rstrip()
    while last_line:
        if last_line.startswith('-'):
            tabs = len( re.match(PATTERN, last_line).group() )
        else:
            tabs = 0
        if tabs < depth:
            break
        node = re.sub(PATTERN, '', last_line.strip())
        if tabs >= depth:
            if parent is not None:
                print "%s: %s" %(parent, node)
                if parent in tree:
                    tree[parent].append(node)
                else:
                    tree[parent] = [ node, ]
            last_line = _recurse_tree(node, tabs+1, source, tree)
    return last_line

または、クラスを使用して状態を保持することもできます。インスタンスから状態を取得する方が簡単です。

class TreeBuilder(object):
    _PATTERN = re.compile('^[-]+')

    def __init__(self, source):
        self.tree = {}
        self.source = source
        self._recurse_tree()

    def _recurse_tree(self, parent=None, depth=0):
         last_line = self.source.readline().rstrip()
         while last_line:
             if last_line.startswith('-'):
                 tabs = len( self._PATTERN.match(last_line).group() )
             else:
                 tabs = 0
             if tabs < depth:
                 break
             node = self._PATTERN.sub('', last_line.strip())
             if tabs >= depth:
                 if parent is not None:
                     print "%s: %s" %(parent, node)
                     if parent in self.tree:
                         self.tree[parent].append(node)
                     else:
                         self.tree[parent] = [ node, ]
                 last_line = self._recurse_tree(node, tabs+1)
         return last_line

次に、これを次のように使用します。

def main():
    inFile = open("data.txt")
    builder = TreeBuilder(inFile)
    pprint.pprint(builder.tree)
于 2012-11-17T18:13:21.990 に答える
1

ここで良い解決策だと思います—クラスを作成し、プライベートクラスメンバーのようにツリーを内部に配置します。

または、この辞書を関数のパラメーターの 1 つとして単純に渡し、再帰中に渡すこともできます。参照渡しなので、常に関数はグローバル変数なしで同じ辞書を使用します。しかし、私はクラスを好むでしょう。

于 2012-11-17T18:13:10.563 に答える
0

関数のデフォルト パラメータとして使用treeします: -

def _recurse_tree(parent, depth, source, tree = None):
    if tree is None:  # needed on first invocation
        tree = {}

最初はtreeパラメーターなしで呼び出し、その後の呼び出しごとに別のパラメーターtreeを追加します。

したがって、メソッド内からの再帰呼び出しは次のようになります。

last_line = _recurse_tree(node, tabs+1, source, tree)
于 2012-11-17T18:13:31.760 に答える