Bagaimana untuk Menghiris Tatasusunan 2D menjadi Subarray 2D yang Lebih Kecil dalam NumPy?

Menghiris Tatasusunan 2D kepada Subarray 2D yang Lebih Kecil

Soalan:
Bolehkah kita membahagikan tatasusunan 2D kepada yang lebih kecil Tatasusunan 2D masuk NumPy?

Contoh:
Ubah tatasusunan 2x4 kepada dua tatasusunan 2x2:

[[1,2,3,4]   ->    [[1,2] [3,4]
 [5,6,7,8]]          [5,6] [7,8]]

Mekanisme:

Daripada mencipta tatasusunan baharu, pendekatan yang lebih baik ialah membentuk semula tatasusunan sedia ada menggunakan reshape() dan tukar paksi menggunakan swapaxes().

Fungsi Berbentuk Blok:

Di bawah ialah pelaksanaan bentuk blok fungsi:

def blockshaped(arr, nrows, ncols):
    """
    Partitions an array into blocks.

    Args:
        arr (ndarray): The original array.
        nrows (int): Number of rows in each block.
        ncols (int): Number of columns in each block.

    Returns:
        ndarray: Partitioned array.
    """
    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))

Demo:

np.random.seed(365)
c = np.arange(24).reshape((4, 6))
print(c)

print(blockshaped(c, 2, 3))

Penyelesaian Alternatif:

blockwise_view SuperBatFish menyediakan pilihan lain, menawarkan susunan blok yang berbeza dan perwakilan berasaskan pandangan.

Atas ialah kandungan terperinci Bagaimana untuk Menghiris Tatasusunan 2D menjadi Subarray 2D yang Lebih Kecil dalam NumPy?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!