numpy的數組列表格式

Question

我怎麼會以這種格式來格式化表格

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

的numpy的數組列表：

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

我試着用兩個for循環

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] 

但這給了我一個索引超出界限的錯誤。我很想了解如何去這個

Answer 1

使用列表中理解和切片的一些想法：

>>> 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]]]