Получение подмножества двумерного массива с учетом индексов центральной точки

Question

Как получить двумерный массив и конкретный элемент с индексами (x, y), как получить квадратный двумерный массив (n x n) с подмножеством с центром в этом элементе?

Я смог реализовать это, только если размер массива подмножеств полностью находится в пределах исходного массива. У меня проблемы, если конкретный элемент находится вблизи краев или углов исходного массива. В этом случае массив подмножеств должен иметь значения nan для элементов вне исходного массива.

Answer 1

Вот как бы я это сделал:

def fixed_size_subset(a, x, y, size):
    '''
    Gets a subset of 2D array given a x and y coordinates
    and an output size. If the slices exceed the bounds 
    of the input array, the non overlapping values
    are filled with NaNs
    ----
    a: np.array
        2D array from which to take a subset
    x, y: int. Coordinates of the center of the subset
    size: int. Size of the output array
    ----       
    Returns:
        np.array
        Subset of the input array
    '''
    o, r = np.divmod(size, 2)
    l = (x-(o+r-1)).clip(0)
    u = (y-(o+r-1)).clip(0)
    a_ = a[l: x+o+1, u:y+o+1]
    out = np.full((size, size), np.nan, dtype=a.dtype)
    out[:a_.shape[0], :a_.shape[1]] = a_
    return out

---

Образцы прогонов:

# random 2D array
a = np.random.randint(1,5,(6,6))

array([[1, 3, 2, 2, 4, 1],
       [1, 3, 1, 3, 3, 2],
       [1, 1, 4, 4, 2, 4],
       [1, 2, 3, 4, 1, 1],
       [4, 1, 4, 2, 3, 4],
       [3, 3, 2, 3, 2, 1]])

---

fixed_size_subset(a, 3, 3, 5)

array([[3., 1., 3., 3., 2.],
       [1., 4., 4., 2., 4.],
       [2., 3., 4., 1., 1.],
       [1., 4., 2., 3., 4.],
       [3., 2., 3., 2., 1.]])

Давайте попробуем привести несколько примеров, в которых нарезанный массив меньше ожидаемого выходного размера:

fixed_size_subset(a, 4, 1, 4)

array([[ 1.,  2.,  3.,  4.],
       [ 4.,  1.,  4.,  2.],
       [ 3.,  3.,  2.,  3.],
       [nan, nan, nan, nan]])

fixed_size_subset(a, 5, 5, 3)

array([[ 3.,  4., nan],
       [ 2.,  1., nan],
       [nan, nan, nan]])

И следующее также будет работать:

fixed_size_subset(a, -1, 0, 3)

array([[ 1.,  3., nan],
       [nan, nan, nan],
       [nan, nan, nan]])

Answer 2

Заполнение массива NaN, а затем выбор соответственно сдвинутого подмассива может помочь.np.pad(arr, (2, 2), "constant", constant_values = np.NaN)