我有一個不可變對象列表，我需要使用屬性作為鍵對其進行排序。對象需要按預先指定的順序排列，而不是簡單的升序/降序。我已經編寫了一些可以做到這一點的代碼，但我不太相信它會在大列表中表現良好（其中“大”完全是任意的）。今天的用例只是小列表（12 項或更少），但我已經活了足夠長的時間，可以期待最終會出現一個大列表。是否有不同的方法、庫或快捷方式我可以利用以使其成為更有效的操作？# setupclass Demo(object): def __init__(self, pk): self.pk = pk def __repr__(self): return f'<Demo: {self.pk}>'# instantiating objects for the sake of demonstration; actual input I'm given is random!objects = [Demo(x) for x in range(10)]# output object should be sorted by pk attribute, with values in this orderordinality = [9, 0, 8, 4, 5, 1, 3, 7, 6, 2]# operation in questionresult = (next(obj for obj in objects if obj.pk == ordinal) for ordinal in ordinality)print(tuple(result))>>> (<Demo: 9>, <Demo: 0>, <Demo: 8>, <Demo: 4>, <Demo: 5>, <Demo: 1>, <Demo: 3>, <Demo: 7>, <Demo: 6>, <Demo: 2>)
3 回答
慕雪6442864
TA貢獻1812條經驗 獲得超5個贊
假設“pk”代表“主鍵”（唯一），那么您可以只使用 dict（具有O(1)成員資格）而不是使用此類對象的列表（具有O(n)成員資格）
objects = [Demo(x) for x in range(10)] # O(n) space
result = [next(obj for obj in objects if obj.pk == ordinal] for ordinal in ordinality) # O(n^2) time
objects = {x: Demo(x) for x in range(10)} # O(n) space
result = [objects[pk] for pk in ordinal] # O(n) time
神不在的星期二
TA貢獻1963條經驗 獲得超6個贊
首先建立一個查找表，然后使用它。把你的 O(n 2 ) 變成 O(n)。
lookup = {obj.pk: obj for obj in objects}
result = [lookup[ordinal] for ordinal in ordinality)
犯罪嫌疑人X
TA貢獻2080條經驗 獲得超4個贊
您可以只使用序數來創建對象。因為序數是你想要開始的順序。
ordinality = [9, 0, 8, 4, 5, 1, 3, 7, 6, 2]
objects = list(map(Demo, ordinality))
#or
objects = [Demo(pk) for pk in ordinality]
添加回答
舉報
0/150
提交
取消