亲宝软件园·资讯

展开

python中的list 查找与过滤方法整合

超级代码搬运工 人气:0

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))#注意此句。

输出是:

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

加载全部内容

相关教程
猜你喜欢
用户评论