問題描述
列表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)問題要求
問題擴(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]
Stay hungry, Stay foolish -- Steve Jobs.
