В Python, как вы сохраняете группировку, когда вы сортируете по одному значению, а затем другому?
Данные выглядят так:
Idx счет группы
5 0,85 Европа
8 0,77 Австралия
12 0,70 С.Америка
13 0,71 Австралия
42 0,82 Европа
45 0,90 Азия
65 0,91 Азия
73 0,72 С.Америка
77 0,84 Азия
Должен выглядеть так:
Idx счет группы
65 0,91 Азия
77 0,84 Азия
45 0,73 Азия
12 0,87 С.Америка
73 0,72 С.Америка
5 0,85 Европа
42 0,82 Европа
8 0,83 Австралия
13 0,71 Австралия
Посмотрите, как у Азии самый высокий балл, и он показывает мне все баллы Азии, затем за ней следует группа, которая имеет 2-й самый высокий балл и так далее? Мне нужно сделать это на Python. Это очень отличается от сортировки по одному элементу, а затем по другому. Пожалуйста помоги. Извините, если этот вопрос излишний. Я едва знаю, как спросить это, не говоря уже о поиске.
У меня был словарь, так что dict = {5:[0,85, Европа],8:[0,77, Австралия]...} И я сделал функцию, которая пыталась проанализировать данные:
def sortResults(dict):
newDict = {}
for k,v in dict.items():
if v[-1] in newDict:
sorDic[v[-1]].append((k,float(v[0]),v[1]))
else:
newDict[v[-1]] = [(k,float(v[0]),v[1])]
for k in newDict.keys():
for resList in newDict[k]:
resList = sorted(resList,key=itemgetter(1),reverse=True)
return sorDic
Он говорит, что поплавок не подлежит подписке... Я просто в замешательстве.
5 ответов
Я бы просто заполнил словарь с максимальным значением для каждой группы, а затем отсортировал бы по максимальному значению в группе с последующим индивидуальным счетом. Как это:
data = [
(5 , 0.85, "Europe"),
(8 , 0.77, "Australia"),
(12, 0.70, "S.America"),
(13, 0.71, "Australia"),
(42, 0.82, "Europe"),
(45, 0.90, "Asia"),
(65, 0.91, "Asia"),
(73, 0.72, "S.America"),
(77, 0.84, "Asia")
]
maximums_by_group = dict()
for indx, score, group in data:
if group not in maximums_by_group or maximums_by_group[group] < score:
maximums_by_group[group] = score
data.sort(key=lambda e: (maximums_by_group[e[2]], e[1]), reverse=True)
for indx, score, group in data:
print indx, score, group
Это дает ожидаемый результат
65 0.91 Asia
77 0.84 Asia
45 0.73 Asia
12 0.87 S.America
73 0.72 S.America
5 0.85 Europe
42 0.82 Europe
8 0.83 Australia
13 0.71 Australia
Самый простой способ сделать это - сбросить данные в список, потому что словари Python не отсортированы. Затем используйте собственный алгоритм timsort в python, который сохраняет прогоны или группировки во время сортировки.
Итак, ваш код будет выглядеть примерно так:
data = [[ 5, 0.85, "Europe"],
[ 8, 0.77, "Australia"],
[12, 0.70, "S.America"],
[13, 0.71, "Australia"],
[42, 0.82, "Europe"],
[45, 0.90, "Asia"],
[65, 0.91, "Asia"],
[73, 0.72, "S.America"],
[77, 0.84, "Asia"]]
data.sort(key=lambda x: x[1], reverse=True)
data.sort(key=lambda x: x[2].upper())
Это даст:
[65, 0.91, 'Asia']
[45, 0.90, 'Asia']
[77, 0.84, 'Asia']
[8, 0.77, 'Australia']
[13, 0.71, 'Australia']
[5, 0.85, 'Europe']
[42, 0.82, 'Europe']
[73, 0.72, 'S.America']
[12, 0.70, 'S.America']
Мне нравятся itertools и оператор:
from itertools import groupby, imap
from operator import itemgetter
def sort_by_max(a_list):
index, score, group = imap(itemgetter, xrange(3))
a_list.sort(key=group)
max_index = dict(
(each, max(imap(index, entries)))
for each, entries in groupby(a_list, group)
)
a_list.sort(key=lambda x:(-max_index[group(x)], -score(x)))
Используется так:
the_list = [
[5, 0.85, 'Europe'],
[8, 0.77, 'Australia'],
[12, 0.87, 'S.America'],
[13, 0.71, 'Australia'],
[42, 0.82, 'Europe'],
[45, 0.90, 'Asia'],
[65, 0.91, 'Asia'],
[73, 0.72, 'S.America'],
[77, 0.84, 'Asia']
]
sort_by_max(the_list)
for each in the_list:
print '{0:2} : {1:<4} : {2}'.format(*each)
дает:
65 : 0.91 : Asia
45 : 0.9 : Asia
77 : 0.84 : Asia
12 : 0.87 : S.America
73 : 0.72 : S.America
5 : 0.85 : Europe
42 : 0.82 : Europe
8 : 0.77 : Australia
13 : 0.71 : Australia
[РЕДАКТИРОВАТЬ]
Если подумать, мне тоже нравится defaultdict а также max:
from collections import defaultdict
def sort_by_max(a_list):
max_index = defaultdict(int)
for index, score, group in a_list:
max_index[group] = max(index, max_index[group])
a_list.sort(key=lambda (index, score, group):(-max_index[group], -score))
Я думаю, что есть лучший способ итерации, чем у меня здесь, но это работает:
from operator import itemgetter
dataset = [
{ 'idx': 5, 'score': 0.85, 'group': 'Europe' },
{ 'idx': 8, 'score': 0.77, 'group': 'Australia' },
{ 'idx': 12, 'score': 0.70, 'group': 'S.America' },
{ 'idx': 13, 'score': 0.71, 'group': 'Australia' },
{ 'idx': 42, 'score': 0.82, 'group': 'Europe' },
{ 'idx': 45, 'score': 0.90, 'group': 'Asia' },
{ 'idx': 65, 'score': 0.91, 'group': 'Asia' },
{ 'idx': 73, 'score': 0.72, 'group': 'S.America' }
]
score_sorted = sorted(dataset, key=itemgetter('score'), reverse=True)
group_score_sorted = []
groups_completed = []
for score in score_sorted:
group_name = score['group']
if not group_name in groups_completed:
groups_completed.append(group_name)
for group in score_sorted:
if group['group'] = group_name:
group_score_sorted.append(group)
#group_score_sorted now contains sorted list
Я думаю, что самый простой способ - сначала разделить по группам, а затем выполнить сортировку в два этапа (первая сортировка по максимуму группы, вторая сортировка по баллам внутри группы).
data = [[ 5, 0.85, "Europe"],
[ 8, 0.77, "Australia"],
[12, 0.70, "S.America"],
[13, 0.71, "Australia"],
[42, 0.82, "Europe"],
[45, 0.90, "Asia"],
[65, 0.91, "Asia"],
[73, 0.72, "S.America"],
[77, 0.84, "Asia"]]
groups = {}
for idx, score, group in data:
try:
groups[group].append((idx, score, group))
except KeyError:
groups[group] = [(idx, score, group)]
for group in sorted((group for group in groups.keys()),
key = lambda g : -max(x[1] for x in groups[g])):
for idx, score, group in sorted(groups[group], key = lambda g : -g[1]):
print idx, score, group
Конечный результат
65 0.91 Asia
45 0.9 Asia
77 0.84 Asia
5 0.85 Europe
42 0.82 Europe
8 0.77 Australia
13 0.71 Australia
73 0.72 S.America
12 0.7 S.America
это отличается от того, что вы предоставили, но для результатов в вашем вопросе, я думаю, у вас есть опечатка, потому что оценка 0.87 за S.America нету чего бы то ни было во входных данных.