Python Sort Array By Two Or Three