Python:在列表中查找

Question

我遇到过这个:

item = someSortOfSelection()
if item in myList:
    doMySpecialFunction(item)

但有时它不适用于我的所有项目,就好像它们在列表中未被识别一样(当它是一个字符串列表时).

这是在列表中找到一个项目的最"Python化"的方式:if x in l:？

Answer 1

至于你的第一个问题:该代码完全正常,如果item等于其中一个元素应该可以工作myList.也许你试图找到一个与其中一个项不完全匹配的字符串,或者你正在使用一个浮点值,它会受到不准确的影响.

至于你的第二个问题:如果在列表中"找到"东西,实际上有几种可能的方法.

检查里面是否有东西

这是您描述的用例:检查列表中是否有内容.如您所知,您可以使用in运算符:

3 in [1, 2, 3] # => True

过滤集合

也就是说,查找序列中满足特定条件的所有元素.您可以使用列表推导或生成器表达式:

matches = [x for x in lst if fulfills_some_condition(x)]
matches = (x for x in lst if x > 6)

后者将返回一个生成器,您可以将其想象为一种惰性列表,只有在您遍历它时才会构建它.顺便说一句,第一个完全等同于

matches = filter(fulfills_some_condition, lst)

在Python 2中,您可以在这里看到高阶函数.在Python 3中,filter不返回列表,而是返回类似于生成器的对象.

找到第一次出现

如果你只想要第一个匹配条件的东西(但你还不知道它是什么),那么使用for循环(可能也使用该else子句,这并不是很有名)是没问题的.你也可以使用

next(x for x in lst if ...)

StopIteration如果没有找到,将返回第一场比赛或加注.或者,您可以使用

next((x for x in lst if ...), [default value])

查找项目的位置

对于列表,index如果您想知道列表中某个元素的位置,那么这个方法有时也很有用:

[1,2,3].index(2) # => 1
[1,2,3].index(4) # => ValueError

但请注意,如果您有重复项,请.index始终返回最低索引:......

[1,2,3,2].index(2) # => 1

如果有重复项并且您想要所有索引,那么您可以使用enumerate():

[i for i,x in enumerate([1,2,3,2]) if x==2] # => [1, 3]

Answer 2

如果要查找一个元素或None使用默认值next,StopIteration如果在列表中找不到该项,则不会引发:

first_or_default = next((x for x in lst if ...), None)

Answer 3

虽然Niklas B.的答案非常全面,但当我们想要在列表中找到一个项目时,获取其索引有时很有用:

next((i for i, x in enumerate(lst) if [condition on x]), [default value])

Answer 4

找到第一次出现

有一个配方itertools:

def first_true(iterable, default=False, pred=None):
    """Returns the first true value in the iterable.

    If no true value is found, returns *default*

    If *pred* is not None, returns the first item
    for which pred(item) is true.

    """
    # first_true([a,b,c], x) --> a or b or c or x
    # first_true([a,b], x, f) --> a if f(a) else b if f(b) else x
    return next(filter(pred, iterable), default)

例如,以下代码查找列表中的第一个奇数:

>>> first_true([2,3,4,5], None, lambda x: x%2==1)
3  

Answer 5

另一种选择：您可以使用 来检查项目是否在列表中if item in list:，但这是 O(n) 的顺序。如果您正在处理大型项目列表，并且您只需要知道某些内容是否属于您的列表，您可以先将列表转换为集合并利用恒定时间集合查找：

my_set = set(my_list)
if item in my_set:  # much faster on average than using a list
    # do something

并非在每种情况下都是正确的解决方案，但在某些情况下，这可能会给您带来更好的性能。

请注意，创建集合set(my_list)也是 O(n)，所以如果您只需要这样做一次，那么这样做不会更快。如果您需要重复检查成员资格，那么在初始集合创建后的每次查找都是 O(1)。

Answer 6

定义和用法

该count()方法返回具有指定值的元素数。

句法

list.count(value)

例子：

fruits = ['apple', 'banana', 'cherry']

x = fruits.count("cherry")

问题示例：

item = someSortOfSelection()

if myList.count(item) >= 1 :

    doMySpecialFunction(item)