Needs a sentence explaining that if `data` is n-dimensional, n > 1, the function is applied along the first axis.