sắp xếp lại các mảng có nhiều mảng

Question

Tôi muốn sắp xếp lại thứ nguyên mảng của tôi. Đoạn mã sau hoạt động nhưng quá chậm.

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] 

Sau này, tôi vectorize các new_im:

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

Điều đó nói rằng, tôi không cần new_im và tôi chỉ cần để có được để new_im_vec. Có cách nào tốt hơn để làm điều này? image_size là khoảng 256.

Answer 1

Kiểm tra rollaxis, một chức năng thay đổi trục xung quanh, cho phép bạn sắp xếp lại mảng theo một lệnh. Nếu im có hình dạng i, j, k

rollaxis(im, 2) 

nên trả về một mảng với hình dạng k, i, j.

Sau đó, bạn có thể làm phẳng mảng, ravel là chức năng rõ ràng cho mục đích này. Đặt tất cả những thứ này lại với nhau, bạn có một lớp lót đẹp:

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 

Điều này sẽ nhanh hơn nhiều vì swapaxes trả về chế độ xem trên mảng, thay vì sao chép các thành phần.

Và tất nhiên nếu bạn muốn bỏ qua new_im, bạn có thể làm điều đó trong một dòng và chỉ còn flatten đang thực hiện bất kỳ thao tác sao chép nào.

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