6

我正在尝试制作一个包含 1 和 0 的所有可能变化的列表。例如，如果我只有两位数字，我想要一个这样的列表：

[[0,0], [0,1], [1,0], [1,1]]

但是，如果我决定有 3 位数字，我希望有一个这样的列表：

[[0,0,0], [0,0,1], [0,1,0], [0,1,1], [1,0,0], [1,0,1], [1,1,0], [1,1,1]]

有人告诉我使用 itertools，但我无法让它以我想要的方式工作。

>>> list(itertools.permutations((range(2))))
[(0, 1), (1, 0)]
>>> [list(itertools.product((range(2))))]
[[(0,), (1,)]]

有没有办法做到这一点？第二个问题，我如何找到关于这样的模块的文档？我只是在这里瞎晃悠

4

3 回答 3

9

itertools.product(.., 重复=n)

>>> import itertools
>>> list(itertools.product((0,1), repeat=3))
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]

Python 模块索引包含标准库模块文档的链接。

于 2013-07-18T12:08:10.257 回答
6

itertools.product()可以接受第二个参数：长度。如您所见，它默认为一个。简单地说，您可以添加repeat=n到您的函数调用：

>>> list(itertools.product(range(2), repeat=3))
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]

要查找文档，您可以使用help(itertools)或只是快速 google（或任何您的搜索引擎）搜索“itertools python”。

于 2013-07-18T12:08:39.300 回答
6

如何找到关于 itertools 的一些信息（除了这里或谷歌），或者任何关于 python 的信息：

python
Python 2.6.6 (r266:84297, Aug 24 2010, 18:46:32) [MSC v.1500 32 bit (Intel)] o
win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import itertools
>>> help(itertools)
Help on built-in module itertools:

NAME
    itertools - Functional tools for creating and using iterators.

FILE
    (built-in)

DESCRIPTION
    Infinite iterators:
    count([n]) --> n, n+1, n+2, ...
    cycle(p) --> p0, p1, ... plast, p0, p1, ...
    repeat(elem [,n]) --> elem, elem, elem, ... endlessly or up to n times

    Iterators terminating on the shortest input sequence:
    izip(p, q, ...) --> (p[0], q[0]), (p[1], q[1]), ...
    izip_longest(p, q, ...) --> (p[0], q[0]), (p[1], q[1]), ...
    ifilter(pred, seq) --> elements of seq where pred(elem) is True
    ifilterfalse(pred, seq) --> elements of seq where pred(elem) is False
    islice(seq, [start,] stop [, step]) --> elements from
           seq[start:stop:step]
    imap(fun, p, q, ...) --> fun(p0, q0), fun(p1, q1), ...
    starmap(fun, seq) --> fun(*seq[0]), fun(*seq[1]), ...
    tee(it, n=2) --> (it1, it2 , ... itn) splits one iterator into n
-- More  --
于 2013-07-18T12:12:59.760 回答