python根据另一个列表的顺序根据属性对对象列表进行排序

Question

我正在使用 Python 列表排序。

我有两个列表:一个是整数列表，另一个是对象列表，第二个对象列表的属性 id 也是一个整数，我想根据 id 属性对对象列表进行排序，按照相同id出现在第一个列表中的顺序，嗯，这是一个例子:

我得到了 a = [1,2,3,4,5]

和b = [o,p,q,r,s]，其中o.id = 2，p.id = 1，q.id = 3，r.id = 5， s.id = 4

我希望我的列表 b 按照其 id 在列表 a 中出现的顺序排序，如下所示:

sorted_b = [p, o, q, s, r]

当然，我可以通过使用嵌套循环来实现:

sorted_b = []
for i in a:
    for j in b:
        if j.id == i:
            sorted_b.append(j)
            break

但这是解决问题的经典丑陋和非 Python 方法，我想知道是否有一种相当简洁的方法来解决这个问题，比如使用 sort 方法，但我不知道如何。

Answer 1

>>> from collections import namedtuple
>>> Foo = namedtuple('Foo', 'name id') # this represents your class with id attribute
>>> a = [1,2,3,4,5]
>>> b = [Foo(name='o', id=2), Foo(name='p', id=1), Foo(name='q', id=3), Foo(name='r', id=5), Foo(name='s', id=4)]
>>> sorted(b, key=lambda x: a.index(x.id))
[Foo(name='p', id=1), Foo(name='o', id=2), Foo(name='q', id=3), Foo(name='s', id=4), Foo(name='r', id=5)]