Python:如何向itertools函数添加参数？

import itertools
from functools import partial

def custom_predicate(x, extra_arg):
    return x < extra_arg  # 假设我们想要过滤掉小于extra_arg的值

# 假设我们有一个列表和一个额外的参数
numbers = [1, 2, 3, 4, 5]
extra_arg_value = 3

# 使用partial固定custom_predicate的extra_arg参数
filtered_numbers = itertools.filterfalse(partial(custom_predicate, extra_arg=extra_arg_value), numbers)

# 打印结果
print(list(filtered_numbers))  # 输出应该是[3, 4, 5]

import itertools

def custom_filterfalse(predicate, iterable, *args, **kwargs):
    def wrapped_predicate(x):
        return predicate(x, *args, **kwargs)
    return itertools.filterfalse(wrapped_predicate, iterable)

def custom_predicate(x, extra_arg):
    return x < extra_arg

numbers = [1, 2, 3, 4, 5]
extra_arg_value = 3

filtered_numbers = custom_filterfalse(custom_predicate, numbers, extra_arg=extra_arg_value)

print(list(filtered_numbers))  # 输出应该是[3, 4, 5]