Python Numpy Array Maximum Value Of A Function