Numpy Array Maximum Element Solution