Pythonの - 私はそうのようなリストにN分木を平らにしたいN子ども（N分木）

Question

を持つツリーをフラット化：

 P 
______|______ 
|  |  | 
C1 C2 C3   =>  [P,C1,C4,C2,C3,C5,C6] 
|   ___|____ 
C4  |  | 
     C5  C6 

これは、ノードクラスです：

class Node(object): 
    def __init__(self, data): 
     self.data = data 
     self.children = [] 
    def add_child(self, obj): 
     self.children.append(obj) 

Answer 1

class Node(object): 
    ... 
    def flatten(self): 
     return [self.data] + sum(
      (c.flatten() for c in self.children), 
      [], 
     ) 

必ずしもわかりやすいわけではありませんが、私はワンライナーのために解決したいと思っていました。

Answer 2

これは私がやったことです。これはうまくいきます。

class Node(object): 
    def __init__(self, data): 
     self.data = data 
     self.children = [] 
    def add_child(self, obj): 
     self.children.append(obj) 
    def flatten(self): 
     out = [self] 
     for child in self.children: 
      out += child.flatten() 
     return out 

私はそのようなreturn [self] + [child.flatten() for child in self.children]として1つのライナー解に到達するためにしばらくの間しようとしたが、それは、ネストされたリストを作成したとして、それが働いたことはありません。もし誰かがもっときれいな方法を知っていれば分かち合ってください