python：展平到列表列表但不多了

发布时间：2020-12-16 23:34:56 所属栏目：Python 来源：网络整理

导读：我有一个嵌套在多个列表层中的列表列表. 可能的输入： [[[[1,2,3],[a,b,c]]]]或[[[1,c]]]或[[[1,3] ]],[[a,c]]] 当我使用flat()时,它会压扁所有不是我想要的东西. [1,3,A,B,C] 我需要的是 [[1,c]] 作为最终输出. 我的平面定义如下 def flat(S): if S == []:

我有一个嵌套在多个列表层中的列表列表.

可能的输入：

[[[[1,2,3],[a,b,c]]]]或[[[1,c]]]或[[[1,3] ]],[[a,c]]]

当我使用flat()时,它会压扁所有不是我想要的东西.

[1,3,A,B,C]

我需要的是

[[1,c]]

作为最终输出.

我的平面定义如下

def flat(S):
    if S == []:
        return S
    if isinstance(S[0],list):
        return flat(S[0]) + flat(S[1:])
    return S[:1] + flat(S[1:])

解决方法

import collections
def is_listlike(x):
    return isinstance(x,collections.Iterable) and not isinstance(x,basestring)

def flat(S):
    result = []
    for item in S:
        if is_listlike(item) and len(item) > 0 and not is_listlike(item[0]):
            result.append(item)
        else:
            result.extend(flat(item))
    return result

tests = [ [[[[1,['a','b','c']]]],[[[1,'c']]],3]],[['a','c']]] ]

for S in tests:
    print(flat(S))

产量

[[1,'c']]
[[1,'c']]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!