python - 如何在 Python 中对 itertools.product 进行排序？

Question

这是 itertools 的产品:

product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy

但我想更改为:--> Ax Bx Cx Dx Ay By Cy Dy 我该怎么做？

Answer 1

更改参数和反转元组的顺序 product产量。它可能比排序更有效，因为它不需要构建中间列表。

from itertools import product

def reorder_product(*a):
    for tup in product(*a[::-1]):
        yield tup[::-1]

例子:

print(*[''.join(t) for t in reorder_product('ABCD', 'xy')])
# Output:
#     Ax Bx Cx Dx Ay By Cy Dy