Как уменьшить выборку двумерного массива путем случайного выбора элементов в подмассивах 2x2?

Question

У меня есть массив 2n x 2m . Я хотел бы сформировать массив n x m , выбрав случайным образом один элемент в 2 x 2 неперекрывающихся подмассивах, которые разбивают мой исходный массив. Каков будет лучший способ сделать это? Есть ли способ избежать двух для петель (по одному на каждое измерение)?

Например, если мой массив

1 2 3 4
5 6 7 8
9 0 1 2
8 5 7 0

тогда есть четыре 2 x 2 подмассива, которые разбивают его:

1 2    3 4
5 6    7 8

9 0    1 2
8 5    7 0

и я хотел бы выбрать по одному элементу в каждом из них, чтобы сформировать новые массивы, такие как

5 3  ,  6 8  ,  2 3
9 2     9 1     0 0  .

Спасибо за ваше время.

Answer 1

Я получил блочную функцию от другого ответа . Этот ответ предполагает, что размер вашего исходного массива соответствует операции.

import numpy as np

def blockshaped(arr, nrows, ncols):
    """
    Return an array of shape (n, nrows, ncols) where
    n * nrows * ncols = arr.size

    If arr is a 2D array, the returned array should look like n subblocks with
    each subblock preserving the "physical" layout of arr.
    """
    h, w = arr.shape
    return (arr.reshape(h//nrows, nrows, -1, ncols)
               .swapaxes(1,2)
               .reshape(-1, nrows, ncols))


arr = np.array([[1,2,3,4],[5,6,7,8],[9,0,1,2],[8,5,7,0]])

#  arr is an 2d array with dimension mxn
m = arr.shape[0]
n = arr.shape[1]

#  define blocksize
block_size = 2

#  divide into sub 2x2 arrays
#  blocks is a (Nx2x2) array
blocks = blockshaped(arr, block_size, block_size)

#  select random elements from each block to form new array
num_blocks = block_size**2
new_arr = blocks[np.arange(num_blocks), np.random.randint(low=0, high=2, size=num_blocks), np.random.randint(low=0, high=2,size=num_blocks)]

print("original array:")
print(arr)

print("random pooled array:")
print(new_arr)

Answer 2

Это можно сделать путем выборки.Вместо выборки каждого квадрата 2x2 мы производим выборку всего ndarray в 4 отдельных ndarray, где один и тот же индекс в этих подмассивах будет указывать в пределах одного и того же квадрата 2x2.И затем мы случайным образом выбираем из этих 4 отдельных ndarray:

# create test dataset
test = np.arange(36).reshape(6,6)

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]])

# Create subsamples from ndarray
samples = np.array([test[::2, ::2], test[1::2, 1::2], test[::2, 1::2], test[1::2, ::2]])
>>> samples
array([[[ 0,  2,  4],
        [12, 14, 16],
        [24, 26, 28]],

       [[ 7,  9, 11],
        [19, 21, 23],
        [31, 33, 35]],

       [[ 1,  3,  5],
        [13, 15, 17],
        [25, 27, 29]],

       [[ 6,  8, 10],
        [18, 20, 22],
        [30, 32, 34]]])

Теперь тот же индекс каждого из этих 4 подвыборок указывает на тот же квадрат 2x2 на исходном ndarray.Нам просто нужно выбрать один и тот же индекс случайным образом:

# Random choice sampling between these 4 subsamples.
select = np.random.randint(4,size=(3,3))
>>> select
array([[2, 2, 1],
       [3, 1, 1],
       [3, 0, 0]])

result = select.choose(samples)
>>> result

array([[ 1,  3, 11],
       [18, 21, 23],
       [30, 26, 28]])