如何将对象附加到作为实例变量的列表?

时间:2015-11-20 21:51:06

标签: python list object hash graph

我的程序有一个由两点确定连接线的方案。我想创建某种图形,并在方案初始化时获得每个点的相邻点。

class Point(object):
    def __init__(self,x,y):
        self.adjacent_points=list()
        self.x=x
        self.y=y
    def __key(self):
        return (self.x, self.y)
    def __hash__(self):
        return hash(self.__key())
    def __eq__(self, other):
        return self.__key()==other.__key()
    def __repr__(self):
        return "Point(%s, %s)" % (self.x, self.y)

class Line(object):
    def __init__(self, point_1, point_2):
        self.point_1=point_1
        self.point_2=point_2

class Scheme(object):
    def __init__(self, element_list):
        self.element_list =element_list
        self.point_list=list()
        self.get_adjacent_points()
    def get_adjacent_points(self):
        self.point_list = list(set(p for l in self.element_list for p in (l.point_1, l.point_2)))
        for l in self.element_list:
            l.point_1.adjacent_points.append(l.point_2)
            l.point_2.adjacent_points.append(l.point_1)

我尝试用两行三点制作简单的方案。

point_list=list()

some_list = list()
some_list.append(Line(Point(0,0), Point(0,1)))
some_list.append(Line(Point(0,1), Point(0,2)))

scheme = Scheme(some_list)

假设坐标(0,1)的点有两个相邻的点:

>>> print(scheme.point_list[0])
Point(0, 1)
>>> print(scheme.point_list[0].adjacent_points)
[Point(0, 0)]

但它只有一个。为什么呢?

2 个答案:

答案 0 :(得分:2)

  

但它只有一个。为什么呢?

由于Point(1, 0) is Point(1, 0)返回false,因此在1, 0旁边有两个单独的点列表。

可能的解决方法:

  1. 重用相同的点对象:

    p01 = Point(0,1)
    
    some_list = [
        Line(Point(0,0), p01))
        Line(p01, Point(0,2)))
    ]
    scheme = Scheme(some_list)
    
  2. 实施__new__,并存储到目前为止找到的所有点的查找表,以便Point(1, 0) is Point(1, 0)返回true。

  3. 不是将数据存储在p.adjacent_points,而是将其存储在scheme.adjacent_points[p]

答案 1 :(得分:1)

问题是第一个列表中Point(0,1)的实例与第二个列表中的实例不同,即使它们共享坐标。您使用了一个集来构造point_list,因此将过滤掉重复的点,但您没有以任何方式使用它来设置相邻点。