2

所以我在python中有这个类

class room (object):

    def __init__(self,number):
        self.room_number=number

    def get_room_num(self):
        return self.room_number

我还有一个类叫House，基本上当我创建house对象时，我还需要为house创建房间对象，房间的数量将由用户选择。

我很难创建动态对象（因为它是由用户指定的，我们将有 1 个房间或 10 个房间），我不能给每个房间对象命名，因为我不知道手头有多少。

所以我试过这样

class House (object):

    def __init__(self,num_of_rooms):
        self.room_holder=[]
        for i in range(num_of_rooms):
            self.room_holder.append(room(i))

    def __iter__(self):
        return iter(self.room_holder)

有没有更好的方法？，如果有，请帮忙

iter函数在那里，所以我可以迭代它，例如

mansion=House(10)

for i in mansion:
    print i.get_room_num()
4

2 回答 2

6 
class Room(object):      # note: class name is Capitalized
    def __init__(self, number):
        self.number = number

    # get_ methods are non-Pythonic.
    # If you need to do some processing to retrieve room number,
    # make it a @property; otherwise, just use the field name

class House(object):
    def __init__(self, num_rooms):
        # I assume you don't want a room 0?
        self.rooms = [Room(i) for i in range(1, num_rooms+1)]
    def __iter__(self):
        return iter(self.rooms)

mansion = House(10)
for room in mansion:
    print room.number
于 2012-06-07T03:43:20.577 回答
3

你那里的东西没有错。有没有更好的办法？好吧，我猜你可以使用列表推导：

class House(object):
  def __init__(self, num_rooms):
    self.rooms=[room(i) for i in xrange(num_rooms)]
  def __iter__(self):
    return iter(self.rooms)

不过，这些本质上是风格上的变化。正如我所说，我认为你所拥有的没有任何问题。

于 2012-06-07T02:38:35.193 回答