Python数组的过滤与筛选技巧

Python 笔记

一、根据条件筛选元素

在Python中，数组的过滤与筛选一般可以通过列表解析或filter()函数实现。当我们需要根据特定的条件来筛选出数组中的元素时，列表解析是更常用的方法。

>>> old_list = [1,2,3,4,5,6,7,8,9]
>>> new_list = [x for x in old_list if x % 2 == 0]
>>> print(new_list)
[2, 4, 6, 8]

在上面的代码中，我们使用了列表解析，判断old_list中的每一个元素是否是偶数，如果是，则添加到new_list中。这样就可以得到一个只包含偶数元素的新数组。

当然，如果数组中的元素非常多，或者根据条件判断比较复杂，我们也可以使用filter()函数来实现筛选操作。

>>> old_list = [1,2,3,4,5,6,7,8,9]
>>> new_list = filter(lambda x: x % 2 == 0, old_list)
>>> print(list(new_list))
[2, 4, 6, 8]

在这段代码中，我们用lambda表达式来定义筛选条件，并将其作为参数传递给filter()函数，然后将结果转换成列表。

二、根据位置筛选元素

在Python中，我们可以直接使用切片操作符“:”来进行位置筛选。

>>> old_list = [1,2,3,4,5,6,7,8,9]
>>> new_list = old_list[3:7] # 筛选出序号为3-6的元素
>>> print(new_list)
[4, 5, 6, 7]

在这段代码中，我们使用了切片操作符“:”，其中3表示起始位置，7表示结束位置（不包含）。通过这样的方式，我们就可以轻松地筛选出数组的一部分。

三、根据值筛选元素

在Python中，我们可以使用index()方法来查找数组中某个元素的位置，然后再使用切片或列表解析来筛选出相应的元素。

>>> old_list = [1,2,3,4,5,6,7,8,9]
>>> index = old_list.index(5) # 查找5在数组中的位置
>>> new_list = old_list[index:]
>>> print(new_list)
[5, 6, 7, 8, 9]

在上面的代码中，我们首先使用index()方法找到数字5在数组中的位置，然后使用切片得到5及其后面的元素。

此外，我们还可以使用列表解析来根据值筛选元素。

>>> old_list = [1,2,3,4,5,6,7,8,9]
>>> new_list = [x for x in old_list if x > 5]
>>> print(new_list)
[6, 7, 8, 9]

在这段代码中，我们使用列表解析得到所有大于5的元素。

四、根据多个条件筛选元素

在Python中，我们可以使用逻辑运算符“and”和“or”来组合多个条件，从而实现根据多个条件筛选元素的功能。

>>> old_list = [1,2,3,4,5,6,7,8,9]
>>> new_list = [x for x in old_list if x > 5 and x % 2 == 0]
>>> print(new_list)
[6, 8]

在这段代码中，我们使用了“and”符号来组合两个条件，得到所有大于5且为偶数的元素。

五、根据正则表达式筛选元素

在Python中，我们可以利用re模块的re.search()函数来使用正则表达式进行数组元素的筛选。

import re

old_list = ['apple', 'banana', 'pear', 'orange']
new_list = [x for x in old_list if re.search('^a', x)]
print(new_list)

在这段代码中，我们使用了正则表达式匹配以字母'a'开头的字符串，并加入到新的数组中。

完整代码示例

# 根据条件筛选元素
old_list = [1,2,3,4,5,6,7,8,9]
new_list = [x for x in old_list if x % 2 == 0]
print(new_list)

# 根据位置筛选元素
old_list = [1,2,3,4,5,6,7,8,9]
new_list = old_list[3:7]
print(new_list)

# 根据值筛选元素
old_list = [1,2,3,4,5,6,7,8,9]
index = old_list.index(5)
new_list = old_list[index:]
print(new_list)

# 根据多个条件筛选元素
old_list = [1,2,3,4,5,6,7,8,9]
new_list = [x for x in old_list if x > 5 and x % 2 == 0]
print(new_list)

# 根据正则表达式筛选元素
import re

old_list = ['apple', 'banana', 'pear', 'orange']
new_list = [x for x in old_list if re.search('^a', x)]
print(new_list)