5

假设我有一个 Person 对象列表，其中包含一个 age 和 room_number 属性，并且我编写了一个 check() 函数，如果 person.age() 和 person.room_number() 都令人满意，则返回 True，否则返回 False。

filter(check, list_of_people_objects)将返回满足以下条件的 Person 对象列表check()

但是，我的问题是，有没有一种方法可以返回每个已批准人员房间号的列表，而无需遍历列表两次，就像不使用列表理解一样？所以过滤，但返回可迭代的更具体的属性。

map(lambda x: x.room_number(), filter(check, list_of_people_objects))

4

3 回答 3

14

其实有两种方式。

  1. itertools

    map(..., itertools.ifilter(..))

  2. 列表理解

    [x.room_number() for x in people if check(x)]

您选择哪种主要是口味问题，但惯例倾向于后者。

于 2012-08-15T09:33:36.840 回答
1

在对象过滤的情况下，您希望对属性子集进行包容性联合以等于一组有限的值，然后执行过滤列表的任何选项（包括列出属性值），您可以使用生成器执行以下操作一条语句（最后一行代码，其余用于显示使用矩阵乘法生成大量对象以生成构造函数参数的指令）

#!/usr/bin/env python
import itertools
import pprint
class myObj(object):
    attr_1 = None
    attr_2 = None
    attr_3 = None
    def __init__(self, at1, at2, at3):
        self.attr_1 = at1
        self.attr_2 = at2
        self.attr_3 = at3
        super(myObj, self).__init__()

    def __repr__(self):
        return '<%s %s>' % (type(self), pprint.pformat(self.__dict__))

objs = itertools.starmap(myObj, itertools.product(iter('val_%d' % (i) for i in
    range(1,4)), repeat=3))

filter_dict = {
    'attr_1' : 'val_1',
    'attr_2' : 'val_2',
    'attr_3' : 'val_3',
}
print(list(result.attr_3 for result in objs if not list(False for pn,cval in
    filter_dict.items() if getattr(result, pn, None) != cval)))
于 2016-01-29T21:53:33.467 回答
0 
class Person():
    def __init__(self,age,room):
        self.age=age
        self.room=room
    def check(self)    :
        if self.room>300 and self.age>15:
            return True
        else:
            return False

输出：

>>> a=Person(20,285)
>>> b=Person(22,990)
>>> c=Person(12,958)
>>> d=Person(18,150)
>>> room=[]
>>> filterd=[]
>>> for x in (a,b,c,d):
    if x.check():
        room.append(x.room)
        filterd.append(x)


>>> room
[990]
>>> filterd
[<__main__.Person object at 0xa94c94c>]
于 2012-08-15T09:43:21.613 回答