make argmax function python code example

Example: argmax python

argmax returns index of max value.