python - 将 str 列表排序为成对的 str，其中一个 str 具有 -R

Question

我有一个字符串列表，如下所示:

["A TB", "A-R TB", "B TB", "B-R TB", "C TB", "C-R TB"...]

但字符串的顺序是随机的。我如何编写一个将元素配对的函数，以便输出如下所示:

[["A TB", "A-R TB"], ["B TB", "B-R TB"], ["C TB", "C-R TB"] ...]

Answer 1

您可以按字典顺序对列表进行排序，然后成对阅读:

def grouped(lst):
    i = iter(sorted(lst))
    return list(zip(i, i))

另一种选择是使用 itertools.groupby() :

def grouped(lst):
    return [list(g) for k, g in groupby(sorted(lst), lambda x: x[0])]

这样你就可以拥有任意大小的组。