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] # 实现问题要求

list.pop( )用法

问题扩展

如果是多重列表,例如 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]

enumerate 用法
eval 用法

Stay hungry, Stay foolish -- Steve Jobs.

推荐阅读更多精彩内容