Làm cách nào để thay đổi định dạng gọn gàng hoạt động?

Question

Tôi có dữ liệu trong một mảng NumPy:

a = np.arange(100) 
a = a.reshape((20,5)) 

Khi tôi gõ

a[:10] 

nó trả

array([[ 0, 1, 2, 3, 4], 
     [ 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14], 
     [15, 16, 17, 18, 19], 
     [20, 21, 22, 23, 24], 
     [25, 26, 27, 28, 29], 
     [30, 31, 32, 33, 34], 
     [35, 36, 37, 38, 39], 
     [40, 41, 42, 43, 44], 
     [45, 46, 47, 48, 49]]) 

Bây giờ tôi đã quyết định thay đổi hình dáng mảng vào mảng 3d.

b = a.reshape((5,4,5)) 

array([[[ 0, 1, 2, 3, 4], 
    [ 5, 6, 7, 8, 9], 
    [10, 11, 12, 13, 14], 
    [15, 16, 17, 18, 19]], 

    [[20, 21, 22, 23, 24], 
    [25, 26, 27, 28, 29], 
    [30, 31, 32, 33, 34], 
    [35, 36, 37, 38, 39]], 

    [[40, 41, 42, 43, 44], 
    [45, 46, 47, 48, 49], 
    [50, 51, 52, 53, 54], 
    [55, 56, 57, 58, 59]], 

    [[60, 61, 62, 63, 64], 
    [65, 66, 67, 68, 69], 
    [70, 71, 72, 73, 74], 
    [75, 76, 77, 78, 79]], 

    [[80, 81, 82, 83, 84], 
    [85, 86, 87, 88, 89], 
    [90, 91, 92, 93, 94], 
    [95, 96, 97, 98, 99]]]) 

Làm cách nào để cắt b để tôi nhận được các giá trị như [: 10]? Tôi đã thử

b[:10,0,:5] 
array([[ 0, 1, 2, 3, 4], 
     [10, 11, 12, 13, 14], 
     [20, 21, 22, 23, 24], 
     [30, 31, 32, 33, 34], 
     [40, 41, 42, 43, 44], 
     [50, 51, 52, 53, 54], 
     [60, 61, 62, 63, 64], 
     [70, 71, 72, 73, 74], 
     [80, 81, 82, 83, 84], 
     [90, 91, 92, 93, 94]]) 

Nhưng không chính xác. Cảm ơn bạn trước!

Answer 1

Khi bạn sử dụng b = a.reshape((5,4,5)) bạn chỉ cần tạo một chế độ xem khác trên cùng một dữ liệu được sử dụng bởi mảng a. (tức là thay đổi thành phần của a sẽ xuất hiện trong b). reshape() không sao chép dữ liệu trong trường hợp này, vì vậy nó hoạt động rất nhanh. Cắt b và cắt a truy cập cùng một bộ nhớ, vì vậy không cần bất kỳ cú pháp nào khác cho một mảng b (chỉ cần sử dụng a[:10]). Nếu bạn có đã tạo bản sao dữ liệu, có lẽ với np.resize() và bị hủy a, chỉ cần định lại lại b: b.reshape((20,5))[:10].

Answer 2

Bằng cách định lại lại (20,5) đến (5,4,5), bạn không thể rút nửa số giá trị đầu tiên ra sao. Bạn không thể chia những 5 hàng vào 2 thậm chí nhóm:

In [9]: b[:2] 
Out[9]: 
array([[[ 0, 1, 2, 3, 4], 
     [ 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14], 
     [15, 16, 17, 18, 19]], 

     [[20, 21, 22, 23, 24], 
     [25, 26, 27, 28, 29], 
     [30, 31, 32, 33, 34], 
     [35, 36, 37, 38, 39]]]) 

In [10]: b[:3] 
Out[10]: 
array([[[ 0, 1, 2, 3, 4], 
     [ 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14], 
     [15, 16, 17, 18, 19]], 

     [[20, 21, 22, 23, 24], 
     [25, 26, 27, 28, 29], 
     [30, 31, 32, 33, 34], 
     [35, 36, 37, 38, 39]], 

     [[40, 41, 42, 43, 44], 
     [45, 46, 47, 48, 49], 
     [50, 51, 52, 53, 54], 
     [55, 56, 57, 58, 59]]]) 

Hàng cuối cùng của a[:10] là ở giữa b[3,:,:].

Lưu ý rằng b[:2] là (2,4,5), 8 hàng của a, chia thành 2 bộ 4.

Bây giờ nếu bạn muốn thực hiện c=a.reshape(4,5,5), sau đó c[:2] sẽ có những người cùng 10 dòng - trong 2 bộ 5. Và c[:2].reshape(10,-1) trông giống như a[:10].