有没有办法把这些 if 写得更好一些？

Question

我需要用 Python 编写这四个 if。请注意它的作用，是在循环中的四种可能状态之间进行更改：1,0 -> 0,1-> -1,0-> 0,-1 并返回到第一个。

if [dx, dy] == [1,0]:
    dx, dy = 0, 1
if [dx, dy] == 0, 1:
    dx, dy = -1, 0
if [dx, dy] == [-1, 0]
    dx, dy = 0, -1
if [dx, dy] == [0, -1]:
    dx, dy = 1, 0

谁能建议我更好/更好的方式来写这个？

Answer 1

dx, dy = -dy, dx

如有疑问，请应用数学。 ;)

Answer 2

马格努斯的建议无疑是对您提出的问题的正确答案，但一般来说，您想使用字典来解决这样的问题：

statemap = {(1, 0): (0, 1), (0, 1): (-1, 0), (-1, 0): (0, -1), (0, -1): (1, 0)}

dx, dy = statemap[dx, dy]

即使在这种情况下，我可以争论使用字典更好，因为很明显有四种状态并且它们重复，但是很难抗拒所有数学的纯粹之美。

顺便说一句，你问题中的代码有其中有一个错误，并且，假设您测试的值是唯一可能的值，相当于：

dx, dy = 1, 0

错误是您需要 elif 作为第二个及后续条件，否则您将继续测试 dx 和 dy 更改后。如果它们是 1 和 0，那么您的所有条件都将成立，并且最终结果相同！如果它们以 0 和 1 开头，那么第二个条件和所有后续条件都将为 true，并且您再次以 1, 0 结束。等等...

Answer 3

只是扩展马格努斯的答案。如果您将 [dx, dy] 想象为一个向量，那么您实际上所做的是 旋转 90 度（或PI/2）。

要计算此值，您可以使用以下转换：

在您的情况下转换为：

x = x * cos(90) - y * sin(90)
y = x * sin(90) + y * cos(90)

由于 sin (90) = 1 和 cos(90) = 0 我们将其简化为：

x, y = -y, x

就这样！

Answer 4

您正在使用的值似乎是一个连续旋转的单位向量 - 换句话说，一个相量。 复数是坐标，因此：

# at initialization
phase = 1
# at the point of modification
phase *= 1j
dx, dy = phase.real, phase.imag

假设我对 dx,dy 值含义的解释是正确的，这为您提供了额外的灵活性，以防以后您想在每个步骤中旋转一些其他量。

Answer 5

虽然我会同意 Magnus 的答案，但这里还有另一种旋转一组值的方法：

def rotate(*states):
    while 1:
        for state in states:
            yield state

for dx, dy in rotate((1, 0), (0, 1), (-1, 0), (0, -1)):
    pass

请注意，在 for dx, dy 循环中的某处应该有一个 break否则它永远不会结束。