2013-07-19 3 views
8

나는이 질문이 실제로 나의 무지 일지라도 What is the equivalent of "zip()" in Python's numpy?과 크게 다르다고 (희망) 생각합니다.python의 zip (*)에 해당하는 numpy는 무엇입니까?

의 다음 I가 있다고 가정 해 봅시다 :

 
[[[ 1, 2], 
    [ 3, 4], 
    [ 5, 6]], 
[[ 7, 8], 
    [ 9, 10], 
    [11, 12]]] 

나는 내가 할 수있는 파이썬에서

 
[[[ 1, 2], 
    [ 7, 8]], 
[[ 3, 4], 
    [ 9, 10]], 
[[ 5, 6], 
    [11, 12]]] 

로 돌려하려면 :

>>> foo 
[[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12]]] 
>>> zip(*foo) 
[([1, 2], [7, 8]), ([3, 4], [9, 10]), ([5, 6], [11, 12])] 

를하지만 내가 어떻게 할 수 numpy 배열 (zip (*)을 사용하지 않고)?

답변

6

실제로 튜플을 반환해야합니까, 아니면 배열을 변형하고 싶습니까?

>>> a 
array([[[ 1, 2], 
     [ 3, 4], 
     [ 5, 6]], 

     [[ 7, 8], 
     [ 9, 10], 
     [11, 12]]]) 

>>> a.swapaxes(0,1) 
array([[[ 1, 2], 
     [ 7, 8]], 

     [[ 3, 4], 
     [ 9, 10]], 

     [[ 5, 6], 
     [11, 12]]]) 
+0

고마워요! 정확히 내가 무엇을 찾고 있었는지. – rhombidodecahedron

관련 문제