Implement Row Extremes, using np.argmax(axis=1) to find the column index of the maximum value in each row.
