Python - 迭代嵌套列表

自從昨天以來，我陷入了一個小而棘手的問題。Python - 迭代嵌套列表

我所擁有的是一個（可能無限）嵌套表是這樣的：

[1,[2,[3,4]]] 
or [[1,2],[3,4]] and so on.

每個級別上的列表包括兩個子表，（我沒有使用元組，因爲該名單可能會得到任意長度在下一步中）現在我想在此列表中的每個可能位置插入一個元素，並返回所有可能插入位置列表的列表。所以，如果我插入5，我的輸出應該是這樣的：

[ [5,[1,[2,[3,4]]]], 
[1,[5,[2,[3,4]]]], 
[1,[2,[5,[3,4]]]], 
[1,[2,[[3,5],4]]], 
[1,[2,[3,[4,5]]]] ]

背景：我試圖通過在每次添加一個類羣來構建系統發育樹。每個分類單位必須插入最合適的位置。

我現在得到的是：

def get_trees(nwklist,newid): 
    if not isinstance(nwklist,list): 
     return [newid,nwklist] 
    else: 
     return [newid,nwklist],[get_trees(nwklist[0],newid),nwklist[1]],[nwklist[0],get_trees(nwklist[1],newid)]

不產生我想要的輸出，但說到有點接近。

([5, [1, [2, [3, 4]]]], 
[[5, 1], [2, [3, 4]]], 
[1, ([5, [2, [3, 4]]], [[5, 2], [3, 4]], [2, ([5, [3, 4]], [[5, 3], 4], [3, [5, 4]])])])

應該有一個簡單的解決方案，可能涉及lambda函數，但我只是沒有看到它。

克里斯托夫

來源

2011-07-27 Christoph

你幾乎肯定會重新發明輪子。 Python中有處理進化樹的軟件包，比如Biopython的Phylo（http://www.biopython.org/wiki/Phylo）或dendropy（http://pypi.python.org/pypi/DendroPy） –

只是挑剔的：你的列表可能具有*任意*深度，但不具有*無限*深度。至少，我不知道如何。 –

「每個分類單位必須插入最合適的位置。」這聽起來像試圖構建一個鄰居加入樹，這是一個不好的計劃。查找更多關於如何最好地構建系統發育樹的文獻，以及爲什麼最簡單的方法也是最差的。 – pyvi

我會使用一個發電機：

def gen_trees(nwklist, newid): 
    yield [newid] + [nwklist] 
    if isinstance(nwklist, list): 
    for i in xrange(len(nwklist)): 
     for l in gen_trees(nwklist[i], newid): 
     yield nwklist[:i] + [l] + nwklist[i+1:] 
    yield [nwklist] + [newid] 

for l in gen_trees([1,[2,[3,4]]] , 5): 
    print l

請注意，這個返回的樹木超過在您的示例所示：

[5, [1, [2, [3, 4]]]] 
[[5, 1], [2, [3, 4]]] 
[[1, 5], [2, [3, 4]]] 
[1, [5, [2, [3, 4]]]] 
[1, [[5, 2], [3, 4]]] 
[1, [[2, 5], [3, 4]]] 
[1, [2, [5, [3, 4]]]] 
[1, [2, [[5, 3], 4]]] 
[1, [2, [[3, 5], 4]]] 
[1, [2, [3, [5, 4]]]] 
[1, [2, [3, [4, 5]]]] 
[1, [2, [[3, 4], 5]]] 
[1, [[2, [3, 4]], 5]] 
[[1, [2, [3, 4]]], 5]

據我所看到的，這符合規定的要求。如果有一些未明確提出的要求我沒有得到（例如，如果每個子列表的第一個元素必須是標量），請澄清，我將更新解決方案。

來源

2011-07-27 12:45:34 NPE

謝謝！這看起來非常好。它不止是我列出的，我忘記提到訂單是不重要的，所以 [1，[[5,2]，[3,4]]] [1，[[2,5]，[3 ，4]]] 基本上是一樣的樹。 – Christoph

如果我刪除第二個yield [nwklist] + [newid]行，我就得到了我想要的。非常感謝！ – Christoph

Python - 迭代嵌套列表

回答

相關問題