zmiana kolejności numpy tablic

Question

Chcę zmienić kolejność wymiarów mojej tablicy numpy. Poniższy fragment kodu działa, ale jest zbyt wolny.

for i in range(image_size): 
    for j in range(image_size): 
     for k in range(3): 
      new_im[k, i, j] = im[i, j, k] 

Po tym, ja wektorować się new_im:

new_im_vec = new_im.reshape(image_size**2 * 3) 

Powiedział, że nie potrzeba new_im a ja tylko potrzeba, aby dostać się new_im_vec. Czy jest lepszy sposób to zrobić? image_size to około 256.

Answer 1

Wyjazd rollaxis, funkcja, która przesuwa się wokół osi, co pozwala uporządkować swoją tablicę w jednym poleceniu. Jeśli im ma kształt i, j, k

rollaxis(im, 2) 

powinien zwracać tablicę z kształt K, I, J.

Po tym można spłaszczyć tablicę, ravel jest jasną funkcją do tego celu. Wyrażając to wszystko razem, masz ładny jedno-liner:

new_im_vec = ravel(rollaxis(im, 2)) 

Answer 2

new_im = im.swapaxes(0,2).swapaxes(1,2) # First swap i and k, then i and j 
new_im_vec = new_im.flatten() # Vectorize 

To powinno być znacznie szybsze, ponieważ swapaxes zwraca widok na tablicy, zamiast kopiować elementy.

I oczywiście, jeśli chcesz pominąć new_im, możesz to zrobić w jednym wierszu i nadal tylko flatten wykonuje jakiekolwiek kopiowanie.

new_im_vec = im.swapaxes(0,2).swapaxes(1,2).flatten()