numpy array to list formatting

Question

How would i format a numpy array of form

data1 = np.array([[0,0,0],[0,1,1],[1,0,1],[1,1,0]])

to a list in this format:

data = [
       [[0,0], [0]],
       [[0,1], [1]],
       [[1,0], [1]],
       [[1,1], [0]]
    ]

I tried using two for loops

for i in range(len(data)):
    for j in range(3):
       if j == 2:
            va[i] = data1[i][j]
       else:            
            sa[i] = data1[i][j]

but that gives me an index out of bounds error. I would love some ideas on how to go about this

@hpaulj va = [] and sa = [] but thank you. I got an answer to my question — Rexfordnana
– Rexfordnana, Commented Dec 8, 2013 at 17:22

Ashwini Chaudhary · Accepted Answer · 2013-12-08 06:04:03Z

5

Use list comprehension and slicing:

>>> data1 = np.array([[0,0,0],[0,1,1],[1,0,1],[1,1,0]])
>>> print [[x[:2].tolist(), x[2:].tolist()] for x in data1]
[[[0, 0], [0]],
 [[0, 1], [1]],
 [[1, 0], [1]],
 [[1, 1], [0]]]

answered Dec 8, 2013 at 6:04

Ashwini Chaudhary

252k60 gold badges478 silver badges519 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

numpy array to list formatting

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related