Yeah, we could, but since we may have to recompute `statistics` if `axis == 1`, I find it slightly clearer that way. This is also how it is done in `fit`.