解决如何在 Python 中的类中赋予属性

Question

我有以下课程：

class Point(object):
    __slots__= ("x","y","z","data","classification")
    def __init__(self,x,y,z,n=None):
        self.x = float(x)
        self.y = float(y)
        self.z = float(z)
        self.data = [self.x,self.y,self.z]

    def set_classification(self,n):
        self.classification = n

p = Point(10,20,30)
p.data
[10.0, 20.0, 30.0]
p.set_classification(0)
p.data
[10.0, 20.0, 30.0]

我有以下问题：

首先，创建Point对象的参数是x,y和z。 Classification 是一个附属参数（有时是文件内的报告其他没有）。如果可能的话，我希望找到一种方法 p.data = [10.0, 20.0, 30.0, 50.0]（设置分类时）或p.data = [10.0, 20.0, 30.0]（未设置分类时）。

其次，我用set_classification

def set_classification(self,n):
      self.classification = n

这是正确的 Python API 风格吗？看了几篇论坛帖子，还是一头雾水。

Answer 1

您可以改用property 来处理分类：

class Point(object):
    __slots__= ("x", "y", "z", "data", "_classification")
    def __init__(self, x, y, z):
        self.x = float(x)
        self.y = float(y)
        self.z = float(z)
        self.data = [self.x,self.y,self.z]

    @property
    def classification(self):
        return getattr(self, '_classification', None)

    @classification.setter
    def classification(self, value):
        self._classification = value
        if value:
            self.data = self.data[:3] + [value]
        else:
            self.data = self.data[:3]

这为classification 属性设置了一个getter 和一个setter；访问classification 属性将转换为第一个被调用的方法。设置 classification 属性在 setter 中进行转换，第二种方法被调用：

p = Point(1, 2, 3)
p.classification = 50  # calls the second `classification` method, the setter, for you.
print p.classification # calls the first `classification` method, the getter.

我在这里选择了classification 属性作为属性，但您也可以将data 设为属性。您将哪一个变成属性取决于使用模式和意图。如果classification 很少更改，则将其设置为属性会更有效。

Answer 2

Getter 和 setter 方法通常不受欢迎。您可以使用属性来实现此目的，但是我建议您始终在返回值中包含 classification，因为脚本的其他部分可能期望 data 始终是 4 项列表。 API 应该是一致的。

这是我使用属性的看法：

class Point(object):
    __slots__= ("x", "y", "z", "data", "classification")
    def __init__(self, x, y, z, n=None):  ## what's n for?
        self.x = float(x)
        self.y = float(y)
        self.z = float(z)

        self.classification = None

    @property
    def data(self):
        if self.classification:
            return [self.x, self.y, self.z, self.classification]
        else:
            return [self.x, self.y, self.z]

p = Point(10, 20, 30)
print p.data   ## [10.0, 20.0, 30.0]
p.classification = 50.0
print p.data   ## [10.0, 20.0, 30.0, 50.0]