numpy ndarray 목록을 여러 목록으로 변환

Question

목록 목록을 5 개로 변환하려는 numpy.ndarray (5,6) (총 크기 (5,6,5)) 목록이 있습니다. 각 목록에는 각 행의 각 numpy.ndarray 정보가 들어 있어야합니다.

이 작업을 수행하는 한 가지 방법은 수동으로 수행하는 것이지만 큰 numpy.ndarrays, 시간이 많이 걸리는 경우 문제가 될 수 있으므로 어떻게 효과적으로 수행 할 수 있습니까?

예 :

>>> import numpy 
>>> a = numpy.zeros((6,5)) 
>>> b = numpy.ones((6,5)) 
>>> c = numpy.ones((6,5))*2 
>>> d = [a,b,c] 
>>> print d 
[array([[ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.]]), array([[ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.]]), array([[ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.]])] 
>>> e = list(d) 
>>> print e 
[array([[ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.]]), array([[ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.], 
     [ 1., 1., 1., 1., 1.]]), array([[ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.], 
     [ 2., 2., 2., 2., 2.]])] 
# not wanted 
>>> print e[0].tolist() 
[[0.0, 0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0, 0.0]] 
#wanted 
>>> [[0.0, 0.0, 0.0, 0.0, 0.0], [1,1,1,1,1], [2, 2, 2, 2, 2], ] 

Answer 1

d

3 개 배열의 목록이다. list(d)은 아무 것도 변경하지 않습니다.

e = np.array(d)은 (3,6,5), 3d 배열을 생성합니다.

In [37]: e=np.array(d) 
In [38]: e.shape 
Out[38]: (3, 6, 5) 
In [39]: e[:,0,:] 
Out[39]: 
array([[0, 0, 0, 0, 0], 
     [1, 1, 1, 1, 1], 
     [2, 2, 2, 2, 2]]) 
In [40]: list(e[:,0,:]) 
Out[40]: [array([0, 0, 0, 0, 0]), array([1, 1, 1, 1, 1]), array([2, 2, 2, 2, 2])] 
In [41]: e[:,0,:].tolist() 
Out[41]: [[0, 0, 0, 0, 0], [1, 1, 1, 1, 1], [2, 2, 2, 2, 2]]