Python Only Keep Rows Together When Sorting