一日一技：在Python中使用reduce

青南

发布于 2019-05-17 14:42:36

5010

发布于 2019-05-17 14:42:36

文章被收录于专栏：未闻Code

如果在Django的model中对多个查询条件使用 或操作，那么代码可以这样写：

from django.db.models import Q
keywords = ['促销', '打折', '限时']
item = Product.objects
if keywords:
    filters = Q(description__contains=keywords[0])
    for keyword in keywords[1:]:
        filters |= Q(description__contains=keyword)
    item = item.filter(filters)

不过这样写代码并不好看，因为要先把第一个参数取出来单独生成一个查询对象。然后再用这个查询对象与后面的参数形成的查询对象取或操作。

对于这种从一个可迭代对象里面依次读取每一个元素，传入到一个函数中，生成的结果再依次与可迭代对象后面的数据进行相同的操作，我们可以使用 reduce。

例如有一个函数 func，它接收两个参数，返回一个参数。现在我们有一个列表， [1,2,3,4,5],想实现：

a = [1, 2, 3, 4, 5]data = func(a[0], a[1])data_2 = func(data, a[2])data_3 = func(data_2, a[3])result = func(data_3, a[4])

那么，代码可以这样修改：

from functools import reduceresult = reduce(func, [1, 2, 3, 4, 5])

所以，对应到Django中不定项或查询，代码可以修改为：

from django.db.models import Qfrom functools import reduce
keywords = ['促销', '打折', '限时']
def func(word_1, word_2):    return Q(description__contains=word_1) | Q(description__contains=word_2)
item = Product.objects
if len(keywords) >= 2:    filters = reduce(func, keywords)else:    filters = Q(description__contains=keyowrds[0]) if keywords else Noneitem = item.filter(filters)

这里需要注意，使用reduce的时候，需要保证它的第二个参数能至少被迭代2次。如果可迭代参数为空列表或者不能迭代的对象，那么就会导致报错。如果只能迭代1次，活着列表里面只有一个元素，那么就会直接返回这个元素，不会被传入函数中。只有当列表里面的元素不小于2个，或者可迭代对象能被迭代的次数不小于2次，reduce才能正常工作。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-05-14，如有侵权请联系 cloudcommunity@tencent.com 删除

mapreduce