Question

我有一个名为Robot的类，它创建了许多类Segment和Pointmass的实例。所有这些实例都有一个变量mass。如何使用相同的变量名mass？

获取对象中所有对象的总质量

我现在做的是什么：

class Robot:
    def __init__(self, massJoint1, massJoint2, massJoint3, massSlide, reachSlide, massArm, lengthArm):
        self.joint1 = Pointmass(massJoint1)
        self.joint2 = Pointmass(massJoint2)
        self.joint3 = Pointmass(massJoint3)

        self.slide = Slide(massSlide, reachSlide)
        self.arm = Arm(massArm, lengthArm)

        self.totalmass = self.joint1.mass + self.joint2.mass + self.joint3.mass + self.slide.mass + self.arm.mass

    def printVars(self):
        attrs = vars(self)
        print(', \n'.join("%s: \t%s" % item for item in attrs.items()))
        print()


class Pointmass:
    def __init__(self, mass):
        self.mass = mass

        self.location = None


class Segment:
    def __init__(self, mass):
        self.mass = mass

        self.start = None
        self.end = None

其中

self.totalmass = self.joint1.mass + self.joint2.mass + self.joint3.mass + self.slide.mass + self.arm.mass

是一条非常长的路线，当Robot获得更多的手臂和关节时真的很不方便......有没有更好的方法来获得总质量？

Answer 1

您不应该将关节存储在单独的属性中。相反，存储一个简单的joints列表;那么你可以简单地将sum与生成器表达式一起使用。

self.totalmass = sum(j.mass for j in self.joints)

在类中添加具有相同名称的所有实例变量

1 个答案: