Question

我有一个对象列表，即

class A:
    int b
    int c

所以

a = A(b=1, c=2)
b = A(b=4, c=2)
...

我有一份这些

的清单

[a, b, c, d, ..., n]

如何将此列表排序为组，其中每个组中的对象在我选择的两个或多个属性之间具有相等性。所以，如果我有

e = A(b=3, c=5)
h = A(b=3, c=5)

这些将被组合成一个列表或一些其他结构（可能设置？）

Answer 1

使用itertools.groupby()。

例如：

from itertools import groupby

class A(object):

    def __init__(self, b, c):
        self.b = b
        self.c = c


items = [A(1, 2), A(1, 2), A(1, 2), A(2, 2), A(3, 4), A(3, 4), A(5, 6)]

groups = groupby(items, lambda a: (a.b, a.c))

for key, group in groups:
    print "Key: %s, Number of items: %s" % (key, len(list(group)))

输出：

Key: (1, 2), Number of items: 3
Key: (2, 2), Number of items: 1
Key: (3, 4), Number of items: 2
Key: (5, 6), Number of items: 1

因此，在这种情况下，关键函数（lambda a: (a.b, a.c)）是将A.b和A.c组合成单个键（两个值的元组）的函数。

group python 2个属性的对象列表

1 个答案: