如何按其第一個元素對 Python 元組元素進行分組？

Python 有一個稱為 defaultdict 的函式，它按元組元素的第一個元素對 Python 元組元素進行分組。

示例

lst = [
   (1, 'Hello', 'World', 112),
   (2, 'Hello', 'People', 42),
   (2, 'Hi', 'World', 200)
]

from collections import defaultdict

d = defaultdict(list)
for k, *v in lst:
   d[k].append(v)
print(d)

輸出

將得到輸出

defaultdict(<class 'list'>, {1: [['Hello', 'World', 112]], 2: [['Hello', 'People', 42], ['Hi', 'World', 200]]})

你可以使用 tuple(d.items()) 方法將此轉換回元組，同時保留分組。

示例

print(tuple(d.items()))

輸出

將得到輸出

((1, [['Hello', 'World', 112]]), (2, [['Hello', 'People', 42], ['Hi', 'World', 200]]))

Arjun Thakur

更新日期： 05-3-2020

565 次瀏覽

開啟職業生涯 生涯

透過完成課程獲得認證

開始