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