itertools产品功能背后的理念

Question

所以基本上我想了解 itertools 中 product() 函数的概念。我的意思是 yield 和 return 之间有什么不同。无论如何可以缩短这段代码吗？

    def product1(*args, **kwds):
        pools = map(tuple, args) * kwds.get('repeat', 1)
        n = len(pools)
        if n == 0:
            yield ()
            return
        if any(len(pool) == 0 for pool in pools):
            return
        indices = [0] * n
        yield tuple(pool[i] for pool, i in zip(pools, indices))
        while 1:
            for i in reversed(range(n)):  # right to left
                if indices[i] == len(pools[i]) - 1:
                    continue
                indices[i] += 1
                for j in range(i+1, n):
                    indices[j] = 0
                yield tuple(pool[i] for pool, i in zip(pools, indices))
                break
            else:
                return

Answer 1

我强烈推荐使用成熟且经过测试的 itertools standard module。作为程序员，重新发明轮子永远不可取。也就是说，我首先要看一下 itertools 中的 product() 函数。

至于不使用itertools()，这个问题本质上是一个笛卡尔积问题（n-permutations with duplicates allowed ).这就是递归帮助我们的地方！一种可能的解决方案如下：

方法体：

result = []
def permutations(alphabet, repeat, total = ''):
    if repeat >= 1:
        for i in alphabet:
            # Add the subsolutions.     
            permutations(alphabet, repeat - 1, total + i)  

    else:
        result.append(total)
    return result

当我们用 permutations()

调用时

示例输出：

permutations('ab', 3) ->
$ ['aaa', 'aab', 'aba', 'abb', 'baa', 'bab', 'bba', 'bbb']
permutations('ab', 3) ->
$ ['aaa', 'aab', 'aac', 'aba', 'abb', 'abc', 'aca', 'acb', 'acc', 'baa',
  'bab', 'bac', 'bba', 'bbb', 'bbc', 'bca', 'bcb', 'bcc', 'caa', 'cab', 
  'cac', 'cba', 'cbb', 'cbc', 'cca', 'ccb', 'ccc']
permutations('ab', 1) ->
$ ['a', 'b']

它是如何工作的？

此方法以递归方式嵌套 for 循环重复次。然后我们累积子解决方案的结果，附加到结果列表。因此，如果我们使用 4 作为我们的重复值，扩展的 iterative 这个问题的痕迹将如下所示：

for i in alphabet:
    for j in alphabet:
        for k in alphabet:
            for l in alphabet:
                result.append(i + j + k + l)

Answer 2

这段代码应该可以完成工作：

bytes = [i for i in range(2**(n))]
AB= []
for obj in bytes:
    t = str(bin(obj))[2:]
    t= '0'*(n-len(t)) + t
    AB.append(t.replace('0','A').replace('1','B'))

n 是想要的字符串大小

Answer 3

首先创建一个包含所有可能排列的列表，这很容易通过对二进制文件求和来实现：

def generate_arrangements(n):
    return [bin(i)[2:].zfill(n) for i in range(2**n)]  # 2**n is number of possible options (A,B) n times

[2:] 对字符串进行切片并从中删除 '0b'，然后 zfill(n) 用 0 补全字符串，直到字符串的长度为 n。

现在分别用A,B替换所有0,1:

arrangements = [arrangement.replace('0', 'A').replace('1', 'B') for arrangement in generate_arrangements(3)]
print(arrangements)
>> ['AAA', 'AAB', 'ABA', 'ABB', 'BAA', 'BAB', 'BBA', 'BBB']

如果你想把所有的东西放在一起，你有：

def generateAB(n):
    arrangements = [bin(i)[2:].zfill(n) for i in range(2**n)]
    return [arrangement.replace('0', 'A').replace('1', 'B') for arrangement in arrangements]

itertools产品功能背后的理念

The Concept Behind itertools's product Function

python

recursion

itertools