问题描述
列表A=[1,[2,3],[4,5,6],7], B=[2,3,4,5,6,7,8], 如何实现B的列表结构和A一样？
问题分析
A 是两重列表，我们只要判断列表 A 的第 i 个元素的长度，如果长度等于1，直接添加 B[i] 即可，否则添加长度为 len(A[i]) 的 B 中对应元素构成的列表即可。
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
输出结果为：
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] # 实现问题要求
问题扩展
如果是多重列表，例如 A=[1,[2,3],[4,[5,6]],7]，上述的方法并不成立，此时，我们可以将A转换为字符串，利用字符串的操作实现多重列表的匹配问题：
stra = list_a.__str__().replace(' ','').split(',')
strb = list_b.__str__().replace(' ','').split(',')
转换后的结果：
['[1', '[2', '3]', '[4', '[5', '6]]', '7]']
['[2', '3', '4', '5', '6', '7', '8]']
将 A 转换为字符串后，我们可以发现多重列表的问题转化为 ' [ ' 或** ' ] '** 的数目问题，我们只需要在 strb 对应元素前或后加入相应数目的 ' [ ' 或 ' ] ' 即可：
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.
