Есть ли способ написать эти ifs лучше?

Question

Мне нужно написать эти четыре if s в Python. Обратите внимание, что он делает, меняется между четырьмя возможными состояниями в цикле: 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, тогда второе и все последующие условия будут истинными, и вы снова получите 1, 0. И так далее ...

Answer 3

Хотя я бы с ответом Магнуса, вот еще один подход для вращения над множеством значений:

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 

Обратите внимание, что там должно быть break где-то в цикле for dx, dy или иначе это будет Никогда не кончается.

Answer 4

Значения, с которыми вы работаете, кажутся единичным вектором, который непрерывно вращается - другими словами, phasor. Complex numbers are coordinates, так:

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

Предполагая, что мое толкование смысла ого, значение д правильно, это дает вам дополнительную гибкость в случае, если выяснится позже, что вы хотите, чтобы повернуть какую-либо другая суммой в каждом шаге.

Answer 5

Просто протягивая Магнус ответ. Если вы представляете [dx, dy] как вектор, то, что вы на самом деле делаете, это rotation из 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 

И вот оно!