https://lacucinadiadine.blogspot.com/2024/10/argmax-of-each-row-or-column-in-scipy.html