Python 找出序列里的符合要求的元素

发表时间

这篇文章要讨论的是,如何用 Python 提取出一个序列中,符合特定条件的所有元素。例如,找出序列里的所有正数、所有以 Wang 结尾的名字、等等


前提

假设序我们要处理的序列为:

range(-5, 5)

也就是

[-5, -4, -3, -2, -1, 0, 1, 2, 3, 4]

注意,本文要处理对象是 iterables,可以是 list、 tuple、set 等sequence,也可以是字符串、甚至是自定义的对象。总之,是可以使用 for 的对象。

下面看看如何提取出序列里的正数:

方法一:循环 Loop

result = []
index = []
for i,x in enumerate(range(-5, 5)):
    if x>0:
        result.append(x)
        index.append(i)

方法二:列表解析 List comprehension

result = [x for x in range(-5, 5) if x>0]
index = [i for i,x in enumerate(range(-5, 5)) if x>0]

方法三:filter 函数

filter 函数和列表解析是等效的,即

filter(function, iterable)

等效于

[item for item in iterable if function(item)]

所有能让 function 得到 True 的值,都被放入结果集。

def pos(x):
    return x>0
result = filter(pos, range(-5,5))

或者使用匿名函数 Lambda 表达式:

result = filter(lambda x:x>0, range(-5,5)) 

本页面内容采用 署名协议 CC-BY 授权。欢迎转载,请保留原文链接


分类

相关文章