Как получить все подматрицы заданного размера в numpy?

Например, x = np.random.randint(low=0, high=10, shape=(6,6)) дает мне массив 6x6 numpy:

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

Как я могу получить список, скажем, всех подматриц 2x3? А не пересекающиеся?

Я мог бы закодировать это сам, но я уверен, что это достаточно распространенная операция, которая уже существует в numpy, я просто не могу ее найти.


python numpy submatrix
person Fequish    schedule 21.09.2015    source источник
comment
Существует ряд SO q и a, касающихся окон в массиве. Некоторые используют это решение.   -  person wwii    schedule 22.09.2015
comment
Дополнительные реализации трюков см. в коде scikit-image и extract_patches из научное обучение   -  person    schedule 24.09.2015

Ответы (1)


arrow_upward
3
arrow_downward

В этом посте приведен общий подход к получению списка подматриц заданной формы. В зависимости от порядка расположения подматриц в строке (в стиле C) или в основном столбце (в стиле fortran), у вас будет два варианта. Вот реализация с np.reshape , np.transpose и np.array_split –

def split_submatrix(x,submat_shape,order='C'):
    p,q = submat_shape      # Store submatrix shape
    m,n = x.shape

    if np.any(np.mod(x.shape,np.array(submat_shape))!=0):
        raise Exception('Input array shape is not divisible by submatrix shape!')

    if order == 'C':
        x4D = x.reshape(-1,p,n/q,q).transpose(0,2,1,3).reshape(-1,p,q)
        return np.array_split(x4D,x.size/(p*q),axis=0)

    elif order == 'F':
        x2D = x.reshape(-1,n/q,q).transpose(1,0,2).reshape(-1,q)
        return np.array_split(x2D,x.size/(p*q),axis=0)

    else:
        print "Invalid output order."
        return x

Пример запуска с измененным вводом образца -

In [201]: x
Out[201]: 
array([[5, 2, 5, 6, 5, 6, 1, 5],
       [1, 1, 8, 4, 4, 5, 2, 5],
       [4, 1, 6, 5, 6, 4, 6, 1],
       [5, 3, 7, 0, 5, 8, 6, 5],
       [7, 7, 0, 6, 5, 2, 5, 4],
       [3, 4, 2, 5, 0, 7, 5, 0]])

In [202]: split_submatrix(x,(3,4))
Out[202]: 
[array([[[5, 2, 5, 6],
         [1, 1, 8, 4],
         [4, 1, 6, 5]]]), array([[[5, 6, 1, 5],
         [4, 5, 2, 5],
         [6, 4, 6, 1]]]), array([[[5, 3, 7, 0],
         [7, 7, 0, 6],
         [3, 4, 2, 5]]]), array([[[5, 8, 6, 5],
         [5, 2, 5, 4],
         [0, 7, 5, 0]]])]

In [203]: split_submatrix(x,(3,4),order='F')
Out[203]: 
[array([[5, 2, 5, 6],
        [1, 1, 8, 4],
        [4, 1, 6, 5]]), array([[5, 3, 7, 0],
        [7, 7, 0, 6],
        [3, 4, 2, 5]]), array([[5, 6, 1, 5],
        [4, 5, 2, 5],
        [6, 4, 6, 1]]), array([[5, 8, 6, 5],
        [5, 2, 5, 4],
        [0, 7, 5, 0]])]
person Divakar    schedule 21.09.2015