Question

所以我有一个很大的嵌套列表，我需要执行以下操作：

查找包含特定元素的列表的数量，如果该数量大于2，则在第二个之后删除所有这些列表。

一个基本的例子：

pgadn_website

尝试：

el = 'ABCfrog'

input = [['ABCdog','a','b'],['ABCfrog','sadg','aseas','ww'],['ABCfrog','dd','ee','ww'],['ABCfrog','dsf','aseas','r','13451v51'],['ABCfrog','3452'],['ABCmouse','xx',],['ABCorange','324213v4']]

output = [['ABCdog','a','b'],['ABCfrog','sadg','aseas','ww'],['ABCfrog','dd','ee','ww'],['ABCmouse','xx',],['ABCorange','324213v4']]

Answer 1

此算法应该有效-它会计数元素在子列表上循环时被查看的次数，并且仅在子列表是包含该元素的前两个子列表之一时才将子列表追加到输出中，或根本不包含该元素。我认为，如果我理解正确，这就是您要记住的逻辑。

el = 'ABCfrog'
inputs = [['ABCdog','a','b'], ['ABCfrog','sadg','aseas','ww'], ['ABCfrog','dd','ee','ww'],  ['ABCfrog','dsf','aseas','r','13451v51'], ['ABCfrog','3452'],['ABCmouse','xx',], ['ABCorange','324213v4']]

outputs = []
count = 0
for i in inputs:
    if el in i and count < 2:
        outputs.append(i)
        count += 1
    elif el not in i:
        outputs.append(i)

此外，请注意，我将您的input变量重命名为inputs，以避免覆盖内置函数input。

Answer 2

这应该有效：

output = []
counter = 0
for sub_list in input:
    if counter >= 2:
        break
    if el in sub_list:
        counter += 1
        output.append(sub_list)
print(output)

删除包含特定元素的最后一个嵌套列表

2 个答案: