如果在所有子列表中的特定位置找不到某個元素，如何刪除子列表？

Question

換句話說，如果發現“f”位於子列表的第4個位置，則返回該子列表，否則，如果未找到“f”，則將其排除。

List = [['a','b','c','d','f'],['a','b','c','d','e'],['a','b','c','d','e'],['a','b','c','f','f'],['a','b']]

如果所有子列表大小相同，我有以下功能。

def Function(SM):
    return filter(lambda x: re.search("f",str(x[4])),List)

    IndexError: list index out of range

 Desired_List = [['a','b','c','d','f'],['a','b','c','f','f']]

由於速度和效率成本的原因，我不願意使用for循環。 有沒有其他同樣快速的選擇？

Answer 1

您可以使用列表理解：

lst = [['a','b','c'], ['a','b','c','d','f'],['a','b','c','d','e'],['a','b','c','d','e'],['a','b','c','f','f'],['a','b']]
lst_desired = [l for l in lst if len(l) >= 5 and l[4] == "f"]
print lst_desired

產量

[['a', 'b', 'c', 'd', 'f'], ['a', 'b', 'c', 'f', 'f']]

Answer 2

>>> li=[['a','b','c','d','f'],['a','b','c','d','e'],['a','b','c','d','e'],['a','b','c','f','f'],['a','b']]
>>> filter(lambda l: l[4:5]==['f'], li)
[['a', 'b', 'c', 'd', 'f'], ['a', 'b', 'c', 'f', 'f']]