python list 查找与过滤方法整合_python list过滤

查找list中是否有某个元素

这个是一个最基础的操作，使用in操作符，如下所示：

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

过滤一个集合

过滤出满足一定条件的所有元素，可以使用列表表达式或者生成器表达式：

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

后者返回一个生成器，你可以将它想象成一个延后列表（lazy list）仅仅当你迭代它的时候才会被创建。顺便说一下，第一个在py2中等价于

matches = filter(fulfills_some_condition, lst)
1

返回一个list。在python3中，filter不会返回一个list，而是一个类似于生成器的对象。

查找第一个匹配项

如果您只想要匹配条件的第一个元素(但是您还不知道它是什么)，那么使用for循环是可以的(也可以使用else子句，但是都不太推荐)。推荐使用：

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

如果没有找到匹配项，则返回第一个匹配项或引发StopIteration异常。或者，您可以使用

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

查找一个item的位置

查找item位置的基础功能可以使用python数组的的index方法

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

但是，请注意，如果您的数组中有重复值，.index总是返回最前面的索引:……

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

如果有重复，你想要所有的索引，你可以使用enumerate()代替:

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

复杂数据查询案例

假如有如下类型的数据

list=[{'id': '2', 'position': [3148.18, 4142.69]},{'id': '1', 'position': [3148.19, 142.69]}]
1

我想根据id来查找，并且获得指定id的position。

list[next(index for (index, d) in enumerate(list1) if d["id"] == '2')]['position']
1

结束