http://www.pythonreader.com/2023/11/numpy-argmax-function-learn-with.html