使用一个列表的值对第二个列表进行分组

Question

假设我有这样的数据：

Owners = ('Bob', 'Bob', 'Bob', 'Jeff', 'Jeff', 'Mary', 'Ana')
Pets = ('Snake', 'Bird', 'Cat', 'Dog', 'Cat', 'Bird', 'Cat')

每一个名字都对应于他们所拥有的动物。 我想创建一个基于所有者的列表列表：

PetsByOwner = [(Snake, Bird, Cat), (Dog, Cat), (Bird), (Cat)]


PetsByOwner= []
for i in Owners:
    if PetsByOwner and PetsByOwner[-1][0] == i:
        PetsByOwner[-1].append(value)
    else:
        PetsByValue.append([value])

我试过了，但它只会破坏主人，而不是宠物。如何连接第二个列表？ 任何帮助都将不胜感激

Answer 1

您可以使用collections.defaultdict：

import collections
Owners = ('Bob', 'Bob', 'Bob', 'Jeff', 'Jeff', 'Mary', 'Ana')
Pets = ('Snake', 'Bird', 'Cat', 'Dog', 'Cat', 'Bird', 'Cat')
d = collections.defaultdict(list)
for a, b in zip(Owners, Pets):
  d[a].append(b)
final_result = {a:tuple(b) for a, b in d.items()}

输出：

{'Bob': ('Snake', 'Bird', 'Cat'), 'Jeff': ('Dog', 'Cat'), 'Mary': ('Bird',), 'Ana': ('Cat',)}

Answer 2

您可以从itertools使用groupby：

from itertools import groupby

Owners = ('Bob', 'Bob', 'Bob', 'Jeff', 'Jeff', 'Mary', 'Ana')
Pets = ('Snake', 'Bird', 'Cat', 'Dog', 'Cat', 'Bird', 'Cat')

print([tuple(v) for g, v in groupby(Pets, key=lambda v, c=iter(Owners): next(c))])

输出：

[('Snake', 'Bird', 'Cat'), ('Dog', 'Cat'), ('Bird',), ('Cat',)]

如果你想给结果添加所有者

print({g: tuple(v) for g, v in groupby(Pets, key=lambda v, c=iter(Owners): next(c))})

印刷品：

{'Bob': ('Snake', 'Bird', 'Cat'), 'Jeff': ('Dog', 'Cat'), 'Mary': ('Bird',), 'Ana': ('Cat',)}