繁体   English   中英

从列表列表中删除项目:pythonic方式

[英]delete items from list of list: pythonic way

我有这种列表列表(只有两个嵌套级别):

my_list = [['A'], ['B'], ['C','D','A','B'], ['E'], ['B', 'F', 'G'], ['H']]

我在my_list列出了要删除的项目:

to_del = ['A','B']

这是我从my_list删除to_del元素的代码的想法:

for i in my_list:
    for d in to_del:
        if d in i:
            i.remove(d)

输出: [[], [], ['C', 'D'], ['E'], ['F', 'G'], ['H']]

在这里我的问题:

  • 你能建议一个更加pythonic /优雅的方式来做同样的事情
  • 你能否提出一个聪明的方法来概括嵌套级别的数量,例如my_list = [ ['A'], ['B'], ['C', ['D', 'E', ['F']], 'G'], ['H'] ]
  • 理想的方法将有一个布尔参数empty_lists来决定是否保留空列表。

尝试列表理解:

my_list = [[x for x in sublist if x not in to_del] for sublist in my_list]

输出:

>>> my_list
[[], [], ['C', 'D'], ['E'], ['F', 'G'], ['H']]

使用嵌套列表推导:

[[y for y in x if y not in to_del] for x in my_list]

使用列表理解和lambda过滤器:

[filter(lambda y: y not in to_del, x) for x in my_list]

尝试任意嵌套列表的一般情况:

def f(e):
    if not isinstance(e,list):
        if e not in to_del:
            return e
    else:
        return filter(None,[f(y) for y in e])

to_del = ['A','B']
my_list= [['A'], ['B',['A','Z', ['C','Z','A']]], ['C','D','A','B'],['E'], ['B','F','G'], ['H']]

>>> f(my_list)
[[['Z', ['C', 'Z']]], ['C', 'D'], ['E'], ['F', 'G'], ['H']]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM