列表理解與 Python 中的模式匹配

Question

我有一個名為columns的列表，我必須根據元素的拆分（前三個）創建一個嵌套列表。

例如，我將這個元素“101 Drive 1 A”分成“101 Drive 1”並組成一個組。

columns = ['101 Drive 1 A','101 Drive 1 B','102 Drive 2 A','102 Drive 2 B','102 Drive 2 C','103 Drive 1 A']

輸出將如下所示：

[
  ['101 Drive 1 A', '101 Drive 1 B'],
  ['102 Drive 2 A', '102 Drive 2 B', '102 Drive 2 C'],
  ['103 Drive 1 A']
]

Answer 1

一種使用collections.defaultdict的方法：

from collections import defaultdict

columns = ['101 Drive 1 A', '101 Drive 1 B', '102 Drive 2 A', '102 Drive 2 B', '102 Drive 2 C', '103 Drive 1 A']

groups = defaultdict(list)
for column in columns:
    key = column[:3]
    groups[key].append(column)

res = list(groups.values())
print(res)

輸出

[['101 Drive 1 A', '101 Drive 1 B'], ['102 Drive 2 A', '102 Drive 2 B', '102 Drive 2 C'], ['103 Drive 1 A']]

一個更健壯的替代方法，適用於字符串開頭的任何數字，是使用正則表達式：

import re

groups = defaultdict(list)
for column in columns:
    key = re.match("\d+", column).group()
    groups[key].append(column)

res = list(groups.values())
print(res)

Answer 2

直接使用itttools的grouby ：

from itertools import groupby

mask = [list(group) for k, group in groupby(columns, lambda s: s.partition(' ')[0])]
print(mask)

給#

[
  ['101 Drive 1 A', '101 Drive 1 B'],
  ['102 Drive 2 A', '102 Drive 2 B', '102 Drive 2 C'],
  ['103 Drive 1 A']
]