在python中列出与彼此元素映射

Posted 2021-04-10

我有一个这样的列表：['A，'B'，'C']，我想要这样的输出

['AB','BC','CA']

我怎样才能得到这个输出。任何帮助赞赏。我将输入列表作为for循环的迭代。谢谢

答案

如果您不关心对之间和对之间的顺序：

import itertools
list(map(''.join, itertools.combinations(['A', 'B', 'C', 'D'], 2)))
['AB', 'AC', 'AD', 'BC', 'BD', 'CD']

如果你照顾：

L = list('ABXD')
N = len(L)
[L[i] + L[(i+d) % N] for d in range(1, N//2 + 1) for i in range(N if 2*d < N else d)]
['AB', 'BX', 'XD', 'DA', 'AX', 'BD']

为了使这个逻辑更加透明，这里是[], ['A'], ['A', 'B'], ... ['A', 'B', 'C', 'D', 'E', 'F']的答案：

[]
[]
['AB']
['AB', 'BC', 'CA']
['AB', 'BC', 'CD', 'DA', 'AC', 'BD']
['AB', 'BC', 'CD', 'DE', 'EA', 'AC', 'BD', 'CE', 'DA', 'EB']
['AB', 'BC', 'CD', 'DE', 'EF', 'FA', 'AC', 'BD', 'CE', 'DF', 'EA', 'FB', 'AD', 'BE', 'CF']

另一答案

使用zip和列表理解，注意lst[1:] + lst[:1]用于将列表滚动一个元素：

lst = ['A','B','C']
[x + y for x, y in zip(lst, lst[1:] + lst[:1])]
# ['AB', 'BC', 'CA']

另一答案

使用伟大的more-itertools包完全懒惰的版本

import itertools
from more_itertools import pairwise

def wrapped_pairwise(it):
    it = iter(it)
    try:
        element = next(it)
    except StopIteration:
        return iter(())
    return (a+b for a, b in pairwise(itertools.chain((element,), it, (element,))))

assert list(wrapped_pairwise("")) == []
assert list(wrapped_pairwise("A")) == ["AA"]
assert list(wrapped_pairwise("ABC")) == ["AB","BC","CA"]
assert list(wrapped_pairwise(["A","B","C"])) == ["AB","BC","CA"]