Для петли в NumPy

Question

Я только начинаю учиться пользоваться numpy.У меня есть этот блок кода, и я собираюсь использовать массивы для его представления.Любые советы о том, как это сделать?

import numpy as np
for i in range(np):
  direction = random.randint(1, 4)
  if direction == NORTH:
    ypositions[i] += 1
  elif direction == SOUTH:
    ypositions[i] -= 1
  elif direction == EAST:
    xpositions[i] += 1
  elif direction == WEST:
    xpositions[i] -= 1

Answer 1

С NumPy вы должны думать с точки зрения целых массивов. Таким образом, применение направленных дельт, основанных на некотором случайном числе, должно работать со всеми координатами одновременно.

Например,

(если предположить, что xpositions и ypositions являются массивами):

directions = np.random.randint(1,4,25)
ypositions += directions == NORTH
ypositions -= directions == SOUTH
xpositions += directions == EAST
xpositions -= directions == WEST