Python:使用OOP和访问方法(自我,其他)

时间:2013-10-28 20:00:41

标签: python oop methods

我想从统一分布中绘制公司类型0和1的位置,并使用Python中的OOP概念计算两种类型之间的距离。

class Firm_loc:

    def __init__(self,type):
        self.type = type
        self.random_location()

    def random_location(self):
        self.location = uniform(0,1), uniform(0,1)

    def distance(self,other):
        xx = (self.location[0] - other.location[0])**2
        yy = (self.location[1] - other.location[1])**2
        return sqrt(xx+yy)

firms = [Firm_loc('Firm_type_0') for i in range(5)]
firms.extend=[Firm_loc('Firm_type_1') for i in range(5)]

问题1:从这里开始,我被困住了。我必须调用距离方法来计算两种公司类型之间的每个距离(即 self 和其他)并将它们保存到列表中。任何帮助将不胜感激。

问题2:是否还有其他更短,更有效的代码,而不是上述代码?

问题3:我正在尝试在OOP中使用“ self 和其他”的概念,因为两种类型的成对计算。如何根据传统的程序方法重写代码?

编辑:

感谢您的答案,但至于距离,我必须有5x5 = 25个距离,如下所示:

Firm_type_0    Firm_type_1    Distance
   f0_0            f0_1       D(f0_0, f0_1)
   f0_0            f1_1       D(f0_0, f1_1)
   f0_0            f2_1       D(f0_0, f2_1)
   f0_0            f3_1       D(f0_0, f3_1)
   ...

我认为这两个答案并没有产生这些距离。

3 个答案:

答案 0 :(得分:1)

我认为你应该将不同的类型存储到不同的列表中。然后迭代这些列表的乘积并计算对的距离。

import itertools
firms = [Firm_loc('Firm_type_0') for i in range(5)]
other_firms = [Firm_loc('Firm_type_1') for i in range(5)]
distances = [i.distance(j) for i, j in itertools.product(firms, other_firms)]

product生成输入可迭代的笛卡尔积。

答案 1 :(得分:1)

我不确定我理解。

你的代码似乎没问题,在类似的东西上没有太多的性能提升空间,然后python对数学来说也不是很好。

如果您想计算公司之间的距离,则需要列表。

firms_a = [Firm_loc('Firm_type_0') for i in range(5)]
firms_b = [Firm_loc('Firm_type_1') for i in range(5)]
all_firms = firms_a + firms_b
firm_distances = [firm_a.distance(firm_b) for firm_a, firm_b in zip(firms_a, firms_b)]
# firm_distance[0] is firms_a[0] distance to firms_b[0] and so on

你在那里使用的“OOP”只是非常基本的语法糖。程序方法可以做到这一点:

Firm_loc.distance(firm_a, firm_b)

以下是:

firm_a.distance(firm_b)

在开始处理继承之前,你更有可能使用语法糖进行结构编程而不是面向对象编程。

修改 为每个type_1获取每个type_0之间的距离:

firms_a = [Firm_loc('Firm_type_0') for i in range(5)]
firms_b = [Firm_loc('Firm_type_1') for i in range(5)]
all_firms = firms_a + firms_b
firm_distances = {
    firm_a:{
        firm_b:firm_a.distance(firm_b) for firm_b in firms_b
    }
    for firm_a in firms_a
}
# Getting the distance of two firms, always A then B
firm_a, firm_b = firms_a[2], firms_b[1]
print(firm_distances[firm_a][firm_b])

# Iterating through distances from a firm A
for firm_b, firm_distance in firm_distances[firm_a].items():
    print(firm_distance)

# Iterating through distances from a firm B
for firm_a in firm_distances:
    print(firm_distances[firm_a][firm_b])

答案 2 :(得分:1)

我不太确定你想要完成什么......这是一个关于发电机的例子,它有点像你想要的,对吧? 5 x 5 ......

firms_a = (Firm_loc('Firm_type_0') for i in range(5))
distances = [[firm.distance(i) for i in (Firm_loc('Firm_type_1') for j in range(5))] for firm in firms_a]

关于OOP问题我认为OdraEncoded提供了很好的答案。你可以把它放到类方法中,例如:

class FirmLoc:
    @classmethod
    def distance(cls, first, second):
        #body of your distance function 
    ...

问题1是什么意思?你被困在哪里?你想要完成什么?