itertool 中列表的条件笛卡尔积

我有四个清单：

LISTA = ['A1', 'A2']
LISTB = ['B1_C', 'B2_D']
LISTC = ['C1', 'C2']
LISTD = ['D1', 'D2']

我想要得到的笛卡尔积LISTA and LISTB，然后根据 B 的值，我想添加 C 的乘积或 D 的乘积。

(A1 B1_C C1)
(A1 B1_C C2)
(A2 B1_C C1)
(A2 B1_C C2)
(A1 B2_D D1)
(A1 B2_D D2)
(A2 B2_D D1)
(A2 B2_D D2)

我可以得到第一部分itertools.product(LISTA, LISTB)，但我一直在寻找itertools https://docs.python.org/2/library/itertools.html对于如何实现第二部分，我不确定最好的方法。建议？

您可以采取product将最后两个列表替换为range相同长度的对象，然后根据项目的最后一个字符选择最后两个列表中的一个LISTB:

from itertools import product

def func(lsta, lstb, lstc, lstd):
    for b, a, i in product(lstb, lsta, range(len(lstc))):
        yield a, b, lstc[i] if b.endswith('C') else lstd[i]

for tup in func(LISTA, LISTB, LISTC, LISTD):          
    print(tup)

('A1', 'B1_C', 'C1')
('A1', 'B1_C', 'C2')
('A2', 'B1_C', 'C1')
('A2', 'B1_C', 'C2')
('A1', 'B2_D', 'D1')
('A1', 'B2_D', 'D2')
('A2', 'B2_D', 'D1')
('A2', 'B2_D', 'D2')