Getting indices of a sorted list of lists in Python