Python:可以交换数学运算符以避免代码重复吗?

Question

我有这个函数(见下文(，看起来它重复了四次。这四个代码片段非常相似，只是在某些数学运算符上有所不同。是否可以使此代码更短？我找不到方法，但我希望能够调用一个函数(在函数中交换代码片段中的"-"、"+"和"<"/">"符号，这样我就不必重复它们了？

法典

def diag_grannar(storlek,troll_position): #funktion som kollar efter diagonala grannar
ny_rad = troll_position[0]
ny_kol = troll_position[1]
diag_grannar = []
# while-loopen letar efter diagonala grannar snett vänster ovanifrån troll_position
while True:
ny_rad -= 1
ny_kol -= 1
if ny_rad >= 1 and ny_kol >= 1:
diag_grannar.append([ny_rad,ny_kol])
else:
ny_rad = troll_position[0]
ny_kol = troll_position[1]
break
# while-loopen letar efter diagonala grannar snett vänster nedanifrån troll_position
while True:
ny_rad += 1
ny_kol -= 1
if ny_rad <= storlek and ny_kol >= 1:
diag_grannar.append([ny_rad,ny_kol])
else:
ny_rad = troll_position[0]
ny_kol = troll_position[1]
break
# while-loopen letar efter diagonala grannar snett höger nedanifrån troll_position
while True:
ny_rad += 1
ny_kol += 1
if ny_rad <= storlek and ny_kol <= storlek:
diag_grannar.append([ny_rad,ny_kol])
else:
ny_rad = troll_position[0]
ny_kol = troll_position[1]
break
# while-loopen letar efter diagonala grannar snett höger ovanifrån troll_position
while True:
ny_rad -= 1
ny_kol += 1
if ny_rad >= 1 and ny_kol <= storlek:
diag_grannar.append([ny_rad,ny_kol])
else:
ny_rad = troll_position[0]
ny_kol = troll_position[1]
break
return diag_grannar

需要明确的是，我所指的代码片段是这些：

while True:
ny_rad -= 1
ny_kol += 1
if ny_rad >= 1 and ny_kol <= storlek:
diag_grannar.append([ny_rad,ny_kol])
else:
ny_rad = troll_position[0]
ny_kol = troll_position[1]
break

此致敬意;

Answer 1

让我们从完全摆脱循环开始：

def diag_grannar(storlek,troll_position):
ny_rad = troll_position[0]
ny_kol = troll_position[1]
diag_grannar = []
diag_grannar.extend([x,y] for x, y in zip(range(ny_rad-1, 0, -1), range(ny_kol-1, 0, -1)))
diag_grannar.extend([x,y] for x, y in zip(range(ny_rad+1, storlek+1), range(ny_kol-1, 0, -1)))
diag_grannar.extend([x,y] for x, y in zip(range(ny_rad+1, storlek+1), range(ny_kol+1, storlek+1)))
diag_grannar.extend([x,y] for x, y in zip(range(ny_rad-1, 0, -1), range(ny_kol+1, storlek+1)))
return diag_grannar

现在，如有必要，我们可以反转范围。range(x-1, 0, -1)与reversed(range(1, x))相同。

def diag_grannar(storlek,troll_position):
ny_rad = troll_position[0]
ny_kol = troll_position[1]
diag_grannar = []
xs = range(1, ny_rad + 1)
ys = range(1, ny_kol + 1)
diag_grannar.extend(list(t) for t in zip(reversed(xs), reversed(ys))
diag_grannar.extend(list(t) for t in zip(xs, reversed(ys))
diag_grannar.extend(list(t) for t in zip(xs, ys))
diag_grannar.extend(list(t) for t in zip(reversed(xs), ys))
return diag_grannar

Answer 2

如果将重复的代码放在函数中，则可以将内置operator模块中的函数作为参数传入，并使用这些函数。

Answer 3

我们可以遍历同一事物的各种参数化......需要注意的是，我们总是可以使用加法，只需添加 1 或 -1(比交换匹配运算符更容易(，而不是交换数学运算符

def diag_grannar1(storlek,troll_position): #funktion som kollar efter diagonala grannar
ny_rad = troll_position[0]
ny_kol = troll_position[1]
diag_grannar = []
def diag_compare(v1, leq):
if leq:
return v1 <= storlek
else:
return v1 >= 1
params = [((-1, -1, False, False)),
((1, -1, True, False)),
((1, 1, True, True)), 
((-1, 1, False, True))]
for v1, v2, leq1, leq2 in params:
while True:
ny_rad += v1
ny_kol += v2
if diag_compare(ny_rad, leq1) and diag_compare(ny_kol, leq2):
diag_grannar.append([ny_rad,ny_kol])
else:
ny_rad = troll_position[0]
ny_kol = troll_position[1]
break
return diag_grannar