https://discountsnowblowersthisinstant.blogspot.com/2024/03/2d-argmax-on-numpy-array.html