numpy数组列出格式

我如何格式化一个numpy形式的数组

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

到这种格式的列表:

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]

但这给了我一个索引超出界限的错误.我会喜欢有关如何解决这个问题的一些想法