Python Array Maximum Value Of A Function