[Numpy-discussion] lists of zeros and ones

Keith Goodman kwgoodman@gmail....
Fri Mar 19 10:21:17 CDT 2010


On Fri, Mar 19, 2010 at 8:17 AM, Joe Kington <jkington@wisc.edu> wrote:
> See itertools.permutations (python standard library)
> e.g.
> In [3]: list(itertools.permutations([1,1,0,0]))
> Out[3]:
> [(1, 1, 0, 0),
>  (1, 1, 0, 0),
>  (1, 0, 1, 0),
>  (1, 0, 0, 1),
>  (1, 0, 1, 0),
>  (1, 0, 0, 1),
>  (1, 1, 0, 0),
>  (1, 1, 0, 0),
>  (1, 0, 1, 0),
>  (1, 0, 0, 1),
>  (1, 0, 1, 0),
>  (1, 0, 0, 1),
>  (0, 1, 1, 0),
>  (0, 1, 0, 1),
>  (0, 1, 1, 0),
>  (0, 1, 0, 1),
>  (0, 0, 1, 1),
>  (0, 0, 1, 1),
>  (0, 1, 1, 0),
>  (0, 1, 0, 1),
>  (0, 1, 1, 0),
>  (0, 1, 0, 1),
>
>
>
>  (0, 0, 1, 1),
>  (0, 0, 1, 1)]
> Hope that helps,
> -Joe

That treats each 1 as distinct. set() solves that:

>> list(set(itertools.permutations([1,1,0,0])))

[(1, 0, 1, 0),
 (1, 1, 0, 0),
 (0, 0, 1, 1),
 (1, 0, 0, 1),
 (0, 1, 1, 0),
 (0, 1, 0, 1)]


More information about the NumPy-Discussion mailing list