我有 2 個清單
in1=[1,1,1,2,2,3,4,4,4,5,5]
in2=['a','b','c','d','e','f','g','h','i','j','k']
我想根據第一個串列中的相同元素對第二個串列進行分組，即
輸出必須是
out=[['a','b','c'],['d','e'],['f'],['g','h','i'],['j','k']]
解釋：第一個串列的前 3 個元素相同，所以我希望將第二個串列的前 3 個元素組合在一起（依此類推）
如果有人可以提供幫助，那就太好了！
~謝謝
uj5u.com熱心網友回復：
只是zip串列，然后itertools進行救援。
from itertools import groupby
in1 = [1,1,1,2,2,3,4,4,4,5,5]
in2 = ['a','b','c','d','e','f','g','h','i','j','k']
result = [[c for _, c in g] for _, g in groupby(zip(in1, in2), key=lambda x: x[0])]
print(result)
# [['a', 'b', 'c'], ['d', 'e'], ['f'], ['g', 'h', 'i'], ['j', 'k']]
非itertools解決方案：
in1 = [1,1,1,2,2,3,4,4,4,5,5]
in2 = ['a','b','c','d','e','f','g','h','i','j','k']
result = [[]]
key = in1[0]
for k, v in zip(in1, in2):
if k != key:
result.append([])
key = k
result[-1].append(v)
print(result)
# [['a', 'b', 'c'], ['d', 'e'], ['f'], ['g', 'h', 'i'], ['j', 'k']]
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/438678.html
標籤：Python python-3.x 列表 分组