python list 查找与过滤方法整合

查找list中是否有某个元素

这个是一个最基础的操作,使用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)

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

matches = filter(fulfills_some_condition, lst)

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

查找第一个匹配项

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

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

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

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

查找一个item的位置

查找item位置的基础功能可以使用python数组的的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]

复杂数据查询案例

假如有如下类型的数据

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

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

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

结束

你可能感兴趣的:(编程技巧,学术,数据可视化,python,数据处理,数据挖掘)