Python 过滤序列元素

目录

• 一、问题
• 二、解决方案
  • • 1. 列表推到
    • 2. 生成器表达式
    • 3. filter() 函数
• 三、讨论

一、问题

过滤序列元素

二、解决方案

1. 列表推到

lst = [1, 4, -5, 10, -7, 2, 3, -1]
print([n for n in lst if n > 0])
# [1, 4, 10, 2, 3]

列表推到占用内存，可以用生成器表达式代替。

2. 生成器表达式

pos = (n for n in lst if n > 0)
print(pos)

for i in pos:
    print(i)

输出：

<generator object <genexpr> at 0x0000027741A098C8>

1
4
10
2
3

过滤规则比较复杂，不能简单的在列表推到或生成器表达式中表达出来，

可以将过滤代码放到一个函数中，然后使用 filter() 函数。

3. filter() 函数

lst = ['1', '2', '3', '-', '4', 'N/A', '5']
def if_int(val):
    for i in val:
        if i.isdigit():
            return True
        else:
            return False
result = list(filter(if_int, lst))
print(result)

输出：

['1', '2', '3', '4', '5']

三、讨论

还有一个过滤工具 itertools.compress()。

以迭代器和布尔选择器作为输入，输出迭代对象中布尔值是 True 的元素。

name = ['wangke', 'wangyan', 'wangying', 'qinlu', 'hongxia']
age = [30, 18, 18, 28, 58]

from itertools import compress

more18 = [n>18 for n in age]
print(more18)
print(list(compress(name, more18)))     # 输出age>18的name

输出：

[True, False, False, True, True]

['wangke', 'qinlu', 'hongxia']