python中按子组的排列

Question

我有一个数据框列的列表，L。

L=[AA ,  AS  ,  AD  , BB  , BC  , C ,  D ]

，我需要所有项目组合，而没有特定的顺序。所以我可以使用我认为的排列吗？

但是每种组合中我只能有一个A和一个B；

所以我需要组合

A=[AA,AS,AD]
B=[BB,BC]
All others=[C,D]

我也不需要全部都存在于输出的排列中。 下面是一些示例组合：

AA,BB,C,D
AA,BB,C
AA
AA,BB

任何人都可以帮忙吗？

Answer 1

我认为您需要一个Cartesian product。您可以使用itertools.product来获得它。由于不需要每个元素，因此可以将None附加到每个列表中，然后再将其删除。

这是一个例子：

from itertools import product
for choices in product(['AA','AS','AD',None],['BB', 'BC', None], ['C', None], ['D', None]):
    print(' '.join(column for column in choices if column))

它输出：

AA BB C D
AA BB C
AA BB D
AA BB
AA BC C D
AA BC C
AA BC D
AA BC
AA C D
AA C
AA D
AA
AS BB C D
AS BB C
AS BB D
AS BB
AS BC C D
AS BC C
AS BC D
AS BC
AS C D
AS C
AS D
AS
AD BB C D
AD BB C
AD BB D
AD BB
AD BC C D
AD BC C
AD BC D
AD BC
AD C D
AD C
AD D
AD
BB C D
BB C
BB D
BB
BC C D
BC C
BC D
BC
C D
C
D