Rumah > pembangunan bahagian belakang > Tutorial Python > Bagaimanakah saya boleh membahagikan tatasusunan NumPy 2D kepada tatasusunan 2D yang lebih kecil?

Bagaimanakah saya boleh membahagikan tatasusunan NumPy 2D kepada tatasusunan 2D yang lebih kecil?

Mary-Kate Olsen
Lepaskan: 2024-11-09 07:44:02
asal
322 orang telah melayarinya

How can I divide a NumPy 2D array into smaller 2D arrays?

Menghiris Tatasusunan 2D kepada Tatasusunan yang Lebih Kecil

Masalah:
Anda ingin memecah dua dimensi ( 2D) Tatasusunan NumPy menjadi tatasusunan 2D yang lebih kecil. Sebagai contoh, anda mungkin mahu menukar tatasusunan 2x4 kepada dua tatasusunan 2x2.

Penyelesaian:
Gabungan bentuk semula dan fungsi swapax membolehkan anda membahagikan tatasusunan anda kepada "blok. " Berikut ialah pelaksanaan Python yang mencapai ini:

def blockshaped(arr, nrows, ncols):
    h, w = arr.shape
    assert h % nrows == 0, f"{h} rows is not evenly divisible by {nrows}"
    assert w % ncols == 0, f"{w} cols is not evenly divisible by {ncols}"

    return (arr.reshape(h//nrows, nrows, -1, ncols)
               .swapaxes(1,2)
               .reshape(-1, nrows, ncols))
Salin selepas log masuk

Dalam penyelesaian ini:

  • nrows dan ncols mentakrifkan dimensi setiap tatasusunan yang lebih kecil.
  • Anda mesti memastikan bahawa tatasusunan input boleh dibahagikan sama rata dengan nilai ini.
  • Fungsi bentuk semula menyusun semula tatasusunan menjadi bongkah.
  • Fungsi swapax menukar paksi untuk mendapatkan bentuk yang diingini.
  • Pembentukan semula akhir meratakan tatasusunan menjadi urutan blok yang lebih kecil.

Contoh:

Pertimbangkan input berikut tatasusunan:

c = np.arange(24).reshape((4, 6))

print(c)

[out]:
[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]
 [12 13 14 15 16 17]
 [18 19 20 21 22 23]]
Salin selepas log masuk

Menggunakan bentuk blok dengan nrows=2 dan ncols=3, anda boleh membahagikan tatasusunan ini ke dalam blok berikut:

print(blockshaped(c, 2, 3))

[out]:
[[[ 0  1  2]
  [ 6  7  8]]

 [[ 3  4  5]
  [ 9 10 11]]

 [[12 13 14]
  [18 19 20]]

 [[15 16 17]
  [21 22 23]]]
Salin selepas log masuk

Demonstrasi ini menggambarkan cara anda boleh menghiris 2D tatasusunan kepada tatasusunan segi empat tepat yang lebih kecil daripada dimensi yang ditentukan.

Atas ialah kandungan terperinci Bagaimanakah saya boleh membahagikan tatasusunan NumPy 2D kepada tatasusunan 2D yang lebih kecil?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

sumber:php.cn
Kenyataan Laman Web ini
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn
Artikel terbaru oleh pengarang
Tutorial Popular
Lagi>
Muat turun terkini
Lagi>
kesan web
Kod sumber laman web
Bahan laman web
Templat hujung hadapan