[英]Python Sort One List According to Another List
我有两个列表,第一个列表是键顺序,第二个列表是元组列表。
colorOrder = ['red', 'blue', 'yellow', 'green']
tupleList = [(111,'red'),(222,'pink'),(333,'green')]
请注意这两个列表不是一对一的关系。 有些颜色不是colorOrder
,对于某些颜色colorOrder
不会出现在tupleList
。 所以它与其他类似的重复问题不同。
我需要根据colorOrder对tupleList进行排序。
我可以使用两个嵌套for循环来解决这个问题,但需要一个更有效的解决方案。
#First sort according to the color order
for aColor in colorOrder:
for aTuple in tupleList:
if aTuple[1] == aColor:
ResultList.append(aTuple)
#Second add the tuples to the ResultList, whose color is not in the colorOrder
for aTuple in tupleList:
if aTuple[1] not in colorOrder:
ResultList.append(aTuple)
首先,我将colorOrder
映射:
colorMap = {c: i for i, c in enumerate(colorOrder)}
现在使用colorMap.get
更轻松地进行排序
sorted(tupleList, key=lambda tup: colorMap.get(tup[1], -1))
这首先将事情不是在地图上。 如果您最后添加它们,只需使用一个非常大的数字:
sorted(tupleList, key=lambda tup: colorMap.get(tup[1], float('inf')))
检查此解决方案:
xx = dict([(x[1],x[0]) for x in enumerate(colorOrder)])
[x[1] for x in sorted([(xx.get(y[1],999),y) for y in tupleList])]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.