Python列表查找五种方法详解：in、index、find、enumerate与NumPy

Python列表查找基础方法概述

Python提供了多种查找功能，主要包括五种方式：in、not in、count、index以及find。其中，in和not in属于保留字，而count和index则是列表对象的方法。

以下示例基于列表a_list = ['a','b','c','hello']进行说明。

对于字符串类型，可以使用find方法来查找子串的位置。

a_list.find('a')

若找到匹配项，find方法返回第一个匹配的位置索引；若未找到，则返回-1。相比之下，若使用index方法查找未找到的元素，程序会直接报错。

index方法的局限性与错误处理

补充知识：在Python中查找包含特定元素的列表索引时，index方法可能会引发错误。

例如，对于列表["foo", "bar", "baz"]及其中元素"bar"，如何获取其索引值（即1）？

一、使用index方法

>>> ["foo", "bar", "baz"].index("bar")

运行结果可能如下所示。

值得注意的是，虽然这是解决该问题最直观的方法，但index作为list API的一部分相对薄弱，许多开发者较少直接使用它。鉴于此答案被广泛引用，需补充相关警告。建议首先查阅文档字符串以了解其限制。

>>> print(list.index.__doc__)
L.index(value, [start, [stop]]) -> integer -- return first index of value.
Raises ValueError if the value is not present.

在实际开发中，我更多使用列表推导式或生成器表达式，因为它们更具通用性。因此，若考虑使用index，建议优先评估这些Python特性。

当元素不在列表中时，index会抛出异常。

具体表现为，若项目不存在，调用index将引发ValueError。

>>> [1, 1].index(2)
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
ValueError: 2 is not in list

如果项目中可能不存在于列表中，应采取以下措施：

首先检查item是否在my_list中（这是一种干净且可读的方法），或者

将index调用包裹在try/except块中捕获ValueError（当搜索列表较长且项通常存在时，这种方法可能更快）。

使用enumerate获取所有匹配索引

二、使用enumerate()函数

大多数答案仅解释如何查找单个索引，但如果项目在列表中多次出现，这些方法无法返回所有索引。此时可使用enumerate()。

for i, j in enumerate(['foo', 'bar', 'baz']):
  if j == 'bar':
    print(i)

index()函数仅返回第一个匹配项，而enumerate()可以返回所有匹配项的索引。

可以通过列表推导式实现：

[i for i, j in enumerate(['foo', 'bar', 'baz']) if j == 'bar']

此外，还有一个使用itertools.count()的小技巧（与enumerate效果几乎相同）：

from itertools import izip as zip, count # izip用于最大化效率

[i for i, j in zip(count(), ['foo', 'bar', 'baz']) if j == 'bar']

对于较大的列表，这种方法比使用enumerate()更有效。

$ python -m timeit -s "from itertools import izip as zip, count" "[i for i, j in zip(count(), ['foo', 'bar', 'baz']*500) if j == 'bar']"
10000 loops, best of 3: 174 usec per loop
$ python -m timeit "[i for i, j in enumerate(['foo', 'bar', 'baz']*500) if j == 'bar']"
10000 loops, best of 3: 196 usec per loop

利用NumPy和zip高效查找

三、使用NumPy库

如果需要获取所有索引，可以使用NumPy。

import numpy as np
 
array = [1, 2, 1, 3, 4, 5, 1]
item = 1
np_array = np.array(array)
item_index = np.where(np_array==item)
print item_index
# Out: (array([0, 2, 6], dtype=int64),)

这是一种清晰且易读的解决方案。

四、使用zip函数

利用zip功能获取所有索引的方法如下：

get_indexes = lambda x, xs: [i for (y, i) in zip(xs, range(len(xs))) if x == y]
 
print get_indexes(2, [1, 2, 3, 4, 5, 6, 3, 2, 3, 2])
print get_indexes('f', 'xsfhhttytffsafweef')

以上关于Python中列表五种查找方法的说明即为全部内容，希望能对大家有所帮助，也欢迎大家继续支持本站。