DOC: Change example to demonstrate function

"a * 0.5" example might as well be new_func(a) directly, it doesn't demonstrate the purpose of apply_along_axis().
Merge pull request #3264 from endolith/patch-1

DOC: Change example to demonstrate function