Как перебрать массив Numpy в пользовательских блоках?

Question

У меня есть массив numpy, по которому я хочу выполнить итерацию. Теперь я знаю функцию nditer (), которая перебирает массив по одной точке за раз. Однако я хочу перебирать массив, беря за раз n точек. Здесь n может находиться в диапазоне от 100 до 200. Любая ссылка на то, как это возможно?

Answer 1

Простое решение будет переделывать.

for chunk in arr.reshape(-1,n):
   ...