using argmax in python code example

Example: argmax python

argmax returns index of max value.