numpy.nditer.remove_axis

метод

nditer.remove_axis(i)

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