Упрощение сортировки данных в Python