您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

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

5b51 2022/1/14 8:23:12 python 字数 1522 阅读 613 来源 www.jb51.cc/python

我有一个嵌套在多个列表层中的列表列表. 可能的输入: [[[[1,2,3],[a,b,c]]]]或[[[1,2,3],[a,b,c]]]或[[[1,2,3] ]],[[a,b,c]]] 当我使用flat()时,它会压扁所有不是我想要的东西. [1,2,3,A,B,C] 我需要的是 [[1,2,3],[a,b,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']]

总结

以上是编程之家为你收集整理的python:展平到列表列表但不多了全部内容,希望文章能够帮你解决python:展平到列表列表但不多了所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶