python列表添加数字_Python实现多重数字列表匹配

问题描述

列表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.