python - 在 python 中重塑一个奇怪的列表

Question

在 Python 中，我有一个这样的列表：

array = 
[[[0,0,1,0],[0,0,0,1],[1,0,0,0],[0,1,0,0]],
[[0,1,0],[1,0,0],[0,0,1],[0,0,1]],
[[0],[1],[0],[1]],
[[1],[1],[0],[1]]]

我想把它改造成：

array = 
[[[0,0,1,0],[0,1,0],[0],[1]],
[[0,0,0,0],[1,0,0],[1],[1]],
[[1,0,0,0],[0,0,1],[0],[0]],
[[0,1,0,0],[0,0,1],[0],[1]]]

有什么建议么？

标签： pythonarrayslistnumpy

Answer 1

IIUC，给定

array = [[[0,0,1,0],[0,0,0,1],[1,0,0,0],[0,1,0,0]],
        [[0,1,0],[1,0,0],[0,0,1],[0,0,1]],
        [[0],[1],[0],[1]],
        [[1],[1],[0],[1]]]

做

import numpy as np
>>> np.array(array).T

array([[list([0, 0, 1, 0]), list([0, 1, 0]), list([0]), list([1])],
       [list([0, 0, 0, 1]), list([1, 0, 0]), list([1]), list([1])],
       [list([1, 0, 0, 0]), list([0, 0, 1]), list([0]), list([0])],
       [list([0, 1, 0, 0]), list([0, 0, 1]), list([1]), list([1])]],