Python-сообщество

import numpy as np
x = input()
a = np.zeros((n, m), dtype=np.int_)
                for i in range(n):
                    for j in range(m):
                        a[i, j] = int(input('A['+str(i+1)+', '+str(j+1)+'] = '))
                 print(np.roll(a, x)) 

[[8 9 5]
 [9 2 9]
 [2 8 9]]
матрица сдвига на 2 позиции
[[8 9 8]
 [9 5 9]
 [2 9 2]]

def matrix_shift(m, shift):
    lr = len(m[0])
    tmp = sum(m, [])
    lm = len(tmp)
    for _ in range(shift):
        tmp.insert(0, tmp.pop(lm-1))
    mtrx = []
    while(tmp):
        row, tmp = tmp[:lr], tmp[lr:]
        mtrx.append(row)
    return mtrx

>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> a_r = a.ravel()
>>> a_r
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
>>> na = np.hstack((a_r[-2:], a_r[:-2]))
>>> na
array([7, 8, 0, 1, 2, 3, 4, 5, 6])
>>> result = na.reshape(3, 3)
>>> result
array([[7, 8, 0],
       [1, 2, 3],
       [4, 5, 6]])

m = [[8, 9, 5],
     [9, 2, 9],
     [2, 8, 9]]
def shift(arr, num):
    return arr[-num:] + arr[:-num]
def flatten(matrix):
    return sum(matrix, [])
def split(arr, num):
    return list(zip(*[iter(arr)]*num))
res = split(shift(flatten(m), 2), len(m[0]))
print res