python中的list 查找与过滤方法整合
作者:超级代码搬运工
这篇文章主要介绍了python中的list 查找与过滤方法整合,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
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过滤列表的技巧
过滤列表可以用循环、列表解析以及filter等。
下面以列表过滤出大于3的元素。
#列表过滤的几种方法 #过滤大于3的元素 #1.利用循环 def filterList1(*lst): b=[] for i in lst: if i>3: b.append(i) return b f1=filterList1(1,2,3,4,5,6) print(f1) #2.利用列表内计算 def filterList2(*lst): b=[i for i in lst if i >3]#列表解析 return b f2=filterList2(1,2,3,4,5,6) print(f2) #3.利用filter def filterList3(*lst): b=filter(lambda x:x>3,lst) return b f3=filterList3(1,2,3,4,5,6) print(list(f3))#注意此句。
输出是:
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。