简单来说,给定一个点A(x,y)和另一个点B(m,n),我需要一个函数,在任何可迭代对象中返回所有中间点的列表[k,z]。
我只对整数点感兴趣,因此不需要浮点数。
我需要最好的Pythonic方法,因为这个“小”函数将被大量运行,并且是更大系统的关键支柱。
编辑:
@roippi,感谢指出有关整数的问题。从下面的代码中可以看到,我尝试跨越x轴并获得相应的y,然后对y执行相同的操作。我的点集不会有任何非离散坐标点,因此暂时可以忽略这个小缺陷。
import itertools
#Vars
origin = {'x':0, 'y':0}
def slope(origin, target):
if target['x'] == origin['x']:
return 0
else:
m = (target['y'] - origin['y']) / (target['x'] - origin['x'])
return m
def line_eqn(origin, target):
x = origin['x']
y = origin['y']
c = -(slope(origin, target)*x - y)
c = y - (slope(origin, target)*x)
#return 'y = ' + str(slope(target)) + 'x + ' + str(c)
m = slope(origin, target)
return {'m':m, 'c':c}
def get_y(x, slope, c):
# y = mx + c
y = (slope*x) + c
return y
def get_x(y, slope, c):
#x = (y-c)/m
if slope == 0:
c = 0 #vertical lines never intersect with y-axis
if slope == 0:
slope = 1 #Do NOT divide by zero
x = (y - c)/slope
return x
def get_points(origin, target):
coord_list = []
#Step along x-axis
for i in range(origin['x'], target['x']+1):
eqn = line_eqn(origin, target)
y = get_y(i, eqn['m'], eqn['c'])
coord_list.append([i, y])
#Step along y-axis
for i in range(origin['y'], target['y']+1):
eqn = line_eqn(origin, target)
x = get_x(i, eqn['m'], eqn['c'])
coord_list.append([x, i])
#return unique list
return list(k for k,_ in itertools.groupby(sorted(coord_list)))
origin = {'x':1, 'y':3}
target = {'x':1, 'y':6}
print get_points(origin, target)
k
和z
都是整数。即使有整数,有些段落只有几个精确的整数对,而其他段落则有很多 - 使得稀疏性高度可变。我认为你还没有完全考虑清楚这个整数问题。 - roippiA = (0, 0)
和B = (17, 19)
,输出应该是什么? - Anton