如何组合元组列表中的唯一值？

Question

我有 2 个列表（来自一个 excel 文件），其中一个包含产品名称，另一个包含数字。

示例 Excel 文件：

这意味着我有 2 个列表 - 使用 xlrd.col_values():

products = ['Product1','Product1','Product1','Product2','Product2','Product2']
values = [1,-1,0,2,4,-1]

我想要的最终结果是这样的：

format = [['Product1', [1,-1,0]],['Product2', [2,4,-1]]]

我曾尝试使用 zip() 来做到这一点：

zip_list = list(zip(products, values))

然而这返回了：

[('Product1', 1), ('Product1', -1), ('Product1', 0), ('Product2', 2), ('Product2', 4), ('Product2', -1)]

有人知道如何获得所需的格式吗？ - 我正在使用python3。

Answer 1

您可以使用字典（理想：collections.defaultdict）来收集数据，然后使用 dict.items() 创建您的格式：

from collections import defaultdict

products = ['Product1','Product1','Product1','Product2','Product2','Product2']
values = [1,-1,0,2,4,-1]

d = defaultdict(list)
# accumulate your data
for prod,val in zip(products,values):
    d[prod].append(val)

print(d)   

# convert dict.items() to your wanted format    
format = [list(i) for i in d.items()] 
print(format)   

输出：

defaultdict(<class 'list'>, {'Product1': [1, -1, 0], 'Product2': [2, 4, -1]})

[['Product1', [1, -1, 0]], ['Product2', [2, 4, -1]]]

使用defaultdict(list) 优于dict.setdefault(key,[]) 或使用try: except: 或测试key in dict，因为它总体上比任何其他方法更快（内置优化）。

独库：

• defaultdict
• zip()

---

您还可以利用itertools.groupby() 对已排序数据（您的已已排序）进行操作以获得相同的结果：

from itertools import groupby

grped = groupby( zip(products,values), lambda x:x[0]) # group by 1st value

l = []
for g in grped:
    l.append([g[0],list(val for _,val in g[1])])      # extract 2nd value from grouping

print(l) # [['Product1', [1, -1, 0]], ['Product2', [2, 4, -1]]]

在此列表中使用 groupby 会创建相同的分区，因为它已排序 - 如果未排序，您会得到不同的结果。

Answer 2

由于字典理解不利于聚合，所以使用线性时间循环：

prods = {}
for item in zip(products, values):
    prod, val = item
    try:
        prods[prod].append(val)
    except KeyError:
        prods[prod] = [val]

# Sample
>>> prods = {}
>>> for item in zip(products, values):
    prod, val = item
    try:
        prods[prod].append(val)
    except KeyError:
        prods[prod] = [val]     
>>> prods
{'Product1': [1, -1, 0], 'Product2': [2, 4, -1]}

我知道您想要[['ProductN', [ ]] 的格式，但我认为字典是更好的选择。