Numpy apply_over_axes