[Numpy-discussion] lists of zeros and ones

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


On Fri, Mar 19, 2010 at 7:53 AM, gerardob <gberbeglia@gmail.com> wrote:
>
> Hello, i would like to produce lists of lists 1's and 0's.
>
> For example, to produce the list composed of:
>
> L = [[1,0,0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]]
>
> I just need to do the following:
>
> n=4
> numpy.eye(n,dtype=int).tolist()
>
> I would like to know a simple way to generate a list containing all the
> lists having two 1's at each element.
>
> Example, n = 4
> L2 = [[1,1,0,0],[1,0,1,0],[1,0,0,1],[0,1,1,0],[0,1,0,1],[0,0,1,1]]
>
> Any ideas?
> Thanks.

Here's the brute force way:

>> for i in range(4):
   ....:     for j in range(i+1, 4):
   ....:         x = np.zeros(4)
   ....:         x[i] = 1
   ....:         x[j] = 1
   ....:         print x
   ....:
   ....:
[ 1.  1.  0.  0.]
[ 1.  0.  1.  0.]
[ 1.  0.  0.  1.]
[ 0.  1.  1.  0.]
[ 0.  1.  0.  1.]
[ 0.  0.  1.  1.]


More information about the NumPy-Discussion mailing list