例如,在指定具有第一优先级的列表之后,我需要为每个唯一名称选择一个具有最小值的列表。
原始嵌套列表:
lst=[[['ahmad','a',5],['ahmad','a',6],['ahmad','c',4],['Emme','b',5],['Emme','b',4]],[['ahmad','b',5],['ahmad','b',6],['ahmad','c',6],['ahmad','c',5],['Meno','c',4],['Emme','b',5],['Moo','b',4],['Moo','a',7],['Moo','a',5]]]
每个列表表示为:['name', 'priority term', value]
。
优先级是“ a”,然后是“ b”,然后是“ c”。
所需结果:
new_lst=[[['ahmad','a',5],['Emme','b',4]],[['ahmad','b',5],['Meno','c',4],['Emme','b',5],['Moo','a',5]]]
更新:
如果列表:
lst=[[['ahmad','red',5,20,'a'],['ahmad','red',6,21,'a'],['ahmad','blue',4,15,'c'],['Emme','red',5,30,'b'],['Emme','red',4,12,'b']],[['ahmad','blue',5,10,'b'],['ahmad','blue',6,13,'b'],['ahmad','blue',6,15,'c'],['ahmad','blue',5,30,'c'],['Meno','green',4,40,'c'],['Emme','green',5,35,'b'],['Moo','red',4,7,'b'],['Moo','red',7,3,'a'],['Moo','red',5,18,'a']]]
每个列表表示为:['name','color',value, trivial number, 'priority term']
。
所需结果:
new_list=[[['ahmad','red',5,20,'a'],['ahmad','blue',4,15,'c'],['Emme','red',4,12,'b']],[['ahmad','blue',5,10,'b'],['Meno','green',4,40,'c'],['Emme','green',5,35,'b'],['Moo','red',5,18,'a']]]
答案 0 :(得分:4)
您可以使用词典来保存优先顺序。然后使用sorted
,然后使用toolz.unique
来排序和删除重复的名称:
from toolz import unique
priority = {v: k for k, v in enumerate('abc')}
def prioritiser(x):
return priority[x[1]], x[2]
res = [list(unique(sorted(sublist, key=prioritiser), key=lambda x: x[0])) \
for sublist in lst]
print(res)
[[['ahmad', 'a', 5], ['Emme', 'b', 4]],
[['Moo', 'a', 5], ['ahmad', 'b', 5], ['Emme', 'b', 5], ['Meno', 'c', 4]]]