問題描述

列表A=[1,[2,3],[4,5,6],7], B=[2,3,4,5,6,7,8], 如何實現(xiàn)B的列表結(jié)構(gòu)和A一樣？

問題分析

A 是兩重列表，我們只要判斷列表 A 的第 i 個元素的長度，如果長度等于1，直接添加 B[i] 即可，否則添加長度為 len(A[i]) 的 B 中對應(yīng)元素構(gòu)成的列表即可。

def copy_struct(lista, listb):
    result = []
    for i in lista:
        if isinstance(i, int):
            result.append(listb.pop(0))
        else:
            result.append([listb.pop(0) for _ in range(len(i))])
    return result

輸出結(jié)果為：

a = [1,[2,3],[4,5,6],7]
b = [2,3,4,5,6,7,8]
print copy_struct(a, b)
>>> [2, [3, 4], [5, 6, 7], 8] # 實現(xiàn)問題要求

list.pop( )用法

問題擴(kuò)展

如果是多重列表，例如 A=[1,[2,3],[4,[5,6]],7]，上述的方法并不成立，此時，我們可以將A轉(zhuǎn)換為字符串，利用字符串的操作實現(xiàn)多重列表的匹配問題：

stra = list_a.__str__().replace(' ','').split(',')
strb = list_b.__str__().replace(' ','').split(',')

轉(zhuǎn)換后的結(jié)果：

['[1', '[2', '3]', '[4', '[5', '6]]', '7]']
['[2', '3', '4', '5', '6', '7', '8]']

將 A 轉(zhuǎn)換為字符串后，我們可以發(fā)現(xiàn)多重列表的問題轉(zhuǎn)化為 ' [ ' 或** ' ] '** 的數(shù)目問題，我們只需要在 strb 對應(yīng)元素前或后加入相應(yīng)數(shù)目的 ' [ ' 或 ' ] ' 即可：

import string
nums = string.digits

def string_number(strrs):
    i = 0
    for _, strr in enumerate(strrs):
         if strr not in nums:
             i += 1
    return i

def structure_b2a(list_a,list_b):
    stra = list_a.__str__().replace(' ','').split(',')
    strb = list_b.__str__().replace(' ','').split(',')
    for i, r in enumerate(stra):    
        if r[0]=='[':
            strb[i] = '[' * string_number(r) + strb[i]  
        if r[-1]==']':
            strb[i] = strb[i] + ']' * string_number(r)

    return eval(strb.__str__()[3:-3].replace("'",''))

算法測試

A=[1,[2,3],[4,[5,6]],7] 
B=[2,3,4,5,6,7,8]
>>> [2, [3, 4], [5, [6, 7]], 8] 

A=[1,[2,3],[4,[[5,6],7]],8]
B=[2,3,4,5,6,7,8,9]
>>> [2, [3, 4], [5, [[6, 7], 8]], 9]

enumerate 用法
eval 用法

Stay hungry, Stay foolish -- Steve Jobs.