如何使用实例作为属性？

Question

让我举一个简单的例子：

如果我想使用Tank 类实例作为汽车的属性。同时，我希望tank_info方法，根据Car类中的不同型号，给出不同的坦克尺寸。

class Car():
    def __init__(self, make, model):
        self.make = make
        self.model = model
        self.tank = Tank()

    def get_car_info(self):
        long_name = self.make + ' ' + self.model
        print('The car is', long_name)

class Tank():
    def __init__(self, tank_size=20):
        self.tank_size = tank_size
    def tank_info(self):
        print('The tank size is', self.tank_size, 'gallons')

my_car = Car('Audi', 'A4')
my_car.get_car_info()
my_car.tank.tank_info()

假设奥迪有 A4、A6 和 A8，它们的油箱尺寸分别为 20、25、30。 我应该如何编写tank_info 方法，让它自动判断它的水箱尺寸？

Answer 1

您可以接受 tank_size 参数到 Car

class Car():
    def __init__(self, make, model, tank_size):
        self.make = make
        self.model = model
        self.tank = Tank(tank_size)

现在你的例子

my_car = Car('Audi', 'A4', 20)
my_car.get_car_info()
my_car.tank.tank_info()

会输出

The car is Audi A4
The tank size is 20 gallons

Answer 2

您必须在实例中存储水箱大小。因此，当您创建 Tank 对象时，您也可以指定大小。这是您创建汽车油箱并因此创建新尺寸的地方。

        self.tank = Tank(tank_size)

以前是

        self.tank = Tank()

这会调用

class Tank():
    def __init__(self, tank_size=20):
        self.tank_size = tank_size

由于您没有通过tank_size，因此默认情况下所有汽车都分配了20尺寸，这是由于tank_size=20而发生的。

---

sizes={
        'Audi': {
                 'A4':25,
                 'A6':30,
                 'A8':35
                },
        'BMW':  {
                 'Series 3':1,
                 'Series 6':2
                }
}

class Car():
    def __init__(self, make, model, tank_size):
        global sizes
        self.make = make
        self.model = model
        self.tank = Tank(sizes[make][model])

    def get_car_info(self):
        long_name = self.make + ' ' + self.model
        print('The car is', long_name)


class Tank():
    def __init__(self, tank_size=20):
        self.tank_size = tank_size
    def tank_info(self):
        print('The tank size is', self.tank_size, 'gallons')

my_car = Car('Audi', 'A4',33)
my_car.get_car_info()
my_car.tank.tank_info()

Answer 3

您可以通过将油箱尺寸映射封装在单独的字典中来避免输入油箱尺寸。

以下示例显示了如何动态构建逻辑。汽车油箱尺寸封装在油箱尺寸字典的字典中，其中外键表示型号，内键表示品牌。如果您想添加更多车型或品牌，只需将它们添加到 car_tank_sizes 字典即可。

另请注意，如果未指定水箱尺寸，则默认为 20。

class Car():
    car_tank_sizes = {'Audi': {'A4': '20', 'A6': 25, 'A8': 30}}

    def __init__(self, make, model):
        self.make = make
        self.model = model
        self.tank = Tank(Car.car_tank_sizes.get(self.make,{}).get(self.model))

    def get_car_info(self):
        long_name = self.make + ' ' + self.model
        print('The car is', long_name)


class Tank():
    def __init__(self, tank_size):
        if (tank_size is None):
            self.tank_size = 20
        else:
            self.tank_size = tank_size

    def tank_info(self):
        print('The tank size is', self.tank_size, 'gallons')


my_car = Car('Audi', 'A6')
my_car.get_car_info()
my_car.tank.tank_info()

下面的替代方法展示了如何在 Tank 本身内封装容器尺寸。

class Car():
    def __init__(self, make, model):
        self.make = make
        self.model = model
        self.tank = Tank(self.make, self.model)

    def get_car_info(self):
        long_name = self.make + ' ' + self.model
        print('The car is', long_name)


class Tank():
    car_tank_sizes = {'Audi': {'A4': '20', 'A6': 25, 'A8': 30}}

    def __init__(self, make, model):
        self.tank_size = Tank.car_tank_sizes.get(make, {}).get(model, 20)

    def tank_info(self):
        print('The tank size is', self.tank_size, 'gallons')


my_car = Car('Audi', 'A4')
my_car.get_car_info()
my_car.tank.tank_info()