numpy.nditer.remove_axis

метод

nditer.remove_axis(i, /)

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