如何获取元组列表的序列[关闭]

Question

鉴于这是输入

[(A,1),(A,2),(B,0),(C,2),(C,3)]

输出应该是：

[[(A,1),(B,0),(C,2)],[(A,2),(B,0),(C,2)],[(A,1),(B,0),(C,3)],[(A,2),(B,0),(C,3)]]

及其排列

Answer 1

尝试使用groupby 和product：

from itertools import groupby, product

input = [("A", 1), ("A", 2), ("B", 0), ("C", 2), ("C", 3)]

groups = groupby(input, key=lambda x: x[0])

output = [list(p) for p in product(*[item for item in (list(i) for g, i in groups)])]

print(output)

输出：

[[('A', 1), ('B', 0), ('C', 2)], [('A', 1), ('B', 0), ('C ', 3)], [('A', 2), ('B', 0), ('C', 2)], [('A', 2), ('B', 0), ( 'C', 3)]]