https://denofgeeksc.pages.dev/posts/how-to-use-the-numpy-argmax-function-in-python/