我想从统一分布中绘制公司类型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)
...
我认为这两个答案并没有产生这些距离。
答案 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是什么意思?你被困在哪里?你想要完成什么?