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

Question

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

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

一个基本的例子：

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']]

尝试：

generic_list = []
for i in input:
    if el in i:
        generic_list.append(i)
number = len(generic_list)

if number>2:
    input.something...

Answer 1

这应该起作用：

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)