Python 实现一个函数找出列表中所有符合条件的元素

Document 对象参考手册 Python3 实例

我们将编写一个 Python 函数，该函数接受一个列表和一个条件函数作为参数，并返回列表中所有符合条件的元素。条件函数将用于判断列表中的每个元素是否满足特定条件。

实例

def find_elements(lst, condition):
    return [x for x in lst if condition(x)]

# 示例条件函数
def is_even(n):
    return n % 2 == 0

# 示例列表
numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

# 使用函数找出所有偶数
even_numbers = find_elements(numbers, is_even)
print(even_numbers)

代码解析：

  1. find_elements 函数接受两个参数：lst 是输入的列表，condition 是一个函数，用于判断列表中的元素是否满足条件。
  2. 使用列表推导式 [x for x in lst if condition(x)] 来遍历列表 lst，并对每个元素 x 应用条件函数 condition(x)。如果条件为真，则该元素被包含在结果列表中。
  3. is_even 是一个示例条件函数，用于判断一个数是否为偶数。
  4. numbers 是一个示例列表，包含一些整数。
  5. 调用 find_elements 函数，传入 numbers 列表和 is_even 条件函数，找出所有偶数并存储在 even_numbers 中。
  6. 最后打印 even_numbers，输出所有符合条件的偶数。

输出结果：

[2, 4, 6, 8, 10]

Document 对象参考手册 Python3 实例