如何将一个两元素方向的元组向左或向右(蛇形人工智能)?

解决方案

import numpy as np

def update_direction (m, move):
    return m = tuple(np.flip((np.array(m) * move) * np.array([1, -1]), 0))

但是,您是否考虑过使用更简单的方向表示法?

例如,只有一个变量 direction 可能的范围是0到3 (0 == North, 1 == East, 2 == South, 3 == West) . 这样当你想转弯时,只需加上一个 turn 值为1或-1。

# Function updates direction
def update_direction(snake, turn):
    snake.direction = (snake.direction + turn) % 4

# turn left
update_direction(snake, -1)

# turn right
update_direction(snake, 1)

然后您可以将此逻辑转换为使蛇最有效的方法。

你可以用线性代数来帮助你解决这个问题。使用numpy,可以将方向向量乘以适当的旋转矩阵:

import numpy as np

if move == -1:
    rot_mat = np.array([[0, -1], [1, 0]])
elif move == 1:
    rot_mat = np.array([[0, 1], [-1, 0]])

这些是数学矩阵,当用你的速度(方向)向量得到点积时,它会给你新的旅行方向向量。

snake.direction = tuple(np.array(snake.direction).dot(rot_mat))

或者如果你想进一步压缩它,你可以用一行代码来完成整个工作。

snake.direction = tuple(np.array(snake.direction).dot(move * np.array([[0, 1], [-1, 0]])))