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

2023-01-04 12:00

短信预约 -IT技能 免费直播动态提醒

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

输出是：