numpy.nditer.remove_axis

метод

nditer.remove_axis(i)

Удаляет ось i из итератора. Требуется, чтобы флаг «multi_index» был включен.