argmax and argmin in python code example

Example: argmax python

argmax returns index of max value.