在python中将带有两个参数的函数传递给filter()

Question

给定以下列表：

DNA_list = ['ATAT', 'GTGTACGT', 'AAAAGGTT']

我想过滤超过 3 个字符的字符串。我通过以下代码实现了这一点：

带有for循环：

long_dna = []
for element in DNA_list:
    length = len(element)
    if int(length) > 3:
        long_dna.append(element)
print long_dna

但我希望我的代码更通用，这样我以后可以过滤任意长度的字符串，所以我使用了一个函数和 for 循环：

def get_long(dna_seq, threshold):
    return len(dna_seq) > threshold

long_dna_loop2 = []
for element in DNA_list:
    if get_long(element, 3) is True:
        long_dna_loop2.append(element)
print long_dna_loop2

我想使用filter() 实现相同的通用性，但我无法实现。如果我使用上面的函数get_long()，当我将它与filter() 一起使用时，我根本无法将参数传递给它。是不可能还是有办法解决？

我的代码filter() 用于特定情况：

def is_long(dna):
        return len(dna) > 3

    long_dna_filter = filter(is_long, DNA_list)

Answer 1

我使用了不同的解决方案，使用内部函数和非本地范围，如下所示。由于我的代码不同，我已经修改了这个原始代码以便理解。

希望这会有所帮助。 :)

 def outerfun():
    charlimit = 3
    def is_long(dna):
        nonlocal charlimit
        return len(dna) > charlimit
    long_dna_filter = filter(is_long, DNA_list)     
    return long_dna_filter

Answer 2

你可以有一个更一般的情况。

由于函数在python中是一个对象，你可以创建另一个函数，它返回你想要的函数。

def f(threshhold):
    def g(x):
        return len(x)>threshhold
    return g #return a function

this_function = f(3)

DNA_list = ['ATAT', 'GTGTACGT', 'AAAAGGTT','AAA','AAAA']
filter(this_function, DNA_list)

output: ['ATAT', 'GTGTACGT', 'AAAAGGTT', 'AAAA']

g 是你真正想要的，f 是创建它的函数。

Answer 3

您始终可以创建一个可调用对象，该可调用对象返回一个适合 filter 进行比较的可调用对象，如下例所示：

def main():
    dna_list = ['A', 'CA', 'TGATGATAC', 'GGGTAAAATC', 'TCG', 'AGGTCGCT', 'TT',
                'GGGTTGGA', 'C', 'TTGGAGGG']
    print('\n'.join(filter(length_at_least(3), dna_list)))


def length_at_least(value):
    return lambda item: len(item) >= value

# length_at_least = lambda value: lambda item: len(item) >= value

if __name__ == '__main__':
    main()

Answer 4

这里有更多使用lambda 的方法。第一个使用默认关键字参数来保存所需的长度。第二个只是将所需的长度嵌入到 lambda 正文中。

#Create a list of strings
s = 'abcdefghi'
data = [s[:i+1] for i in range(len(s))]
print data

thresh = 3
print filter(lambda seq, n=thresh: len(seq) > n, data)

print filter(lambda seq: len(seq) > 5, data)

输出

['a', 'ab', 'abc', 'abcd', 'abcde', 'abcdef', 'abcdefg', 'abcdefgh', 'abcdefghi']
['abcd', 'abcde', 'abcdef', 'abcdefg', 'abcdefgh', 'abcdefghi']
['abcdef', 'abcdefg', 'abcdefgh', 'abcdefghi']

在第一个例子中你也可以这样做：

print filter(lambda seq, n=3: len(seq) > n, data)

同样，在第二个示例中，您可以将文字 5 替换为局部（或全局）变量，例如：

thresh = 5
print filter(lambda seq: len(seq) > thresh, data)

Answer 5

你需要使用filter()吗？为什么不使用更 Pythonic 的列表理解？

例子：

>>> DNA_list = ['ATAT', 'GTGTACGT', 'AAAAGGTT']
>>> threshold = 3
>>> long_dna = [dna_seq for dna_seq in DNA_list if len(dna_seq) > threshold]
>>> long_dna
['ATAT', 'GTGTACGT', 'AAAAGGTT']

>>> threshold = 4
>>> [dna_seq for dna_seq in DNA_list if len(dna_seq) > threshold]
['GTGTACGT', 'AAAAGGTT']

此方法的优点是可以轻松地将其转换为生成器，该生成器可以根据您的应用程序提供改进的内存和执行，例如如果你有很多 DNA 序列，并且你想遍历它们，将它们实现为一个列表将一次性消耗大量内存。等效的生成器只需将方括号 [] 替换为圆括号 ()：

>>> long_dna = (dna_seq for dna_seq in DNA_list if len(dna_seq) > threshold)
<generator object <genexpr> at 0x7f50de229cd0>
>>> list(long_dna)
['GTGTACGT', 'AAAAGGTT']

在 Python 2 中，filter() 不能提供这种性能改进，因为它返回一个列表。在 Python 3 中，filter() 返回一个更类似于生成器的过滤器对象。

Answer 6

您正在尝试做的事情被称为partial function application：您有一个带有多个参数的函数（在本例中为 2），并且想要从它派生一个固定一个或多个参数的函数，然后您可以传递给filter。

某些语言（尤其是函数式语言）“内置”了此功能。在 python 中，您可以使用 lambdas 来执行此操作（如其他人所示），或者您可以使用functools library。特别是functools.partial：

partial() 用于部分函数应用程序，它“冻结”函数参数和/或关键字的某些部分，从而生成具有简化签名的新对象。例如，partial() 可用于创建行为类似于 int() 函数的可调用对象，其中基本参数默认为两个：

>>> from functools import partial
>>> basetwo = partial(int, base=2)
>>> basetwo.__doc__ = 'Convert base 2 string to an int.'
>>> basetwo('10010')
18

所以你可以这样做：

filter(functools.partial(get_long, treshold=13), DNA_list)

Answer 7

你可以让is_long返回一个函数，它可以接受dna，像这样

>>> def is_long(length):
...     return lambda dna: len(dna) > length
... 

然后在filter中使用，像这样

>>> filter(is_long(3), DNA_list)
['ATAT', 'GTGTACGT', 'AAAAGGTT']
>>> filter(is_long(4), DNA_list)
['GTGTACGT', 'AAAAGGTT']

---

注意：不要使用is 运算符来比较布尔值或数字。而是尽可能地依赖数据的真实性。所以，在你的情况下，你可以像这样写你的第二个版本

if get_long(element, 3):
    long_dna_loop2.append(element)

引用programming recommendations in PEP-8，

不要使用 == 将布尔值与 True 或 False 进行比较。

 Yes:   if greeting:
 No:    if greeting == True:
 Worse: if greeting is True:

Answer 8

使用lambda 提供阈值，如下所示：

filter(lambda seq: get_long(seq, 3),
       dna_list)