zl程序教程

您现在的位置是:首页 >  后端

当前栏目

Python: 序列: 过滤序列元素

Python序列 元素 过滤
2023-09-14 09:07:34 时间

问题: 你有一个数据序列,想利用一些规则从中提取出需要的值或者是缩短序列

answer:

eg1:列表推导

最简单的过滤序列元素的方法就是使用列表推导。比如:
>>> mylist = [1, 4, -5, 10, -7, 2, 3, -1]
>>> [n for n in mylist if n > 0]
[1, 4, 10, 2, 3]
>>> [n for n in mylist if n < 0]
[-5, -7, -1]

 

eg2:生成器表达式迭代

使用列表推导的一个潜在缺陷就是如果输入非常大的时候会产生一个非常大的结果集,占用大量内存。如果你对内存比较敏感,那么你可以使用生成器表达式迭代产生过滤的元素。比如:
>>> pos = (n for n in mylist if n > 0)
>>> pos
<generator object <genexpr> at 0x1006a0eb0>
>>> for x in pos:
... print(x)
...
1

4
10
2

3

eg3: filter()

有时候,过滤规则比较复杂,不能简单的在列表推导或者生成器表达式中表达出来。比如,过滤的时候需要处理一些异常或者其他复杂情况。可以将过滤代码放到一个函数中,然后使用内建的filter() 函数

>>>values = ['1', '2', '-3', '-', '4', 'N/A', '5']
>>>def is_int(val):
...  try:
...    x = int(val)
...    return True
...  except ValueError:
...    return False
>>>ivals = list(filter(is_int, values))
>>>print(ivals)
# Outputs ['1', '2', '-3', '4', '5']
filter() 函数创建了一个迭代器,因此如果你想得到一个列表的话,就得像示例那样使用list() 去转换。