python中实现回旋矩阵的代码

# numpy导入  先在终端下输入pip install numpy，下载第三方包numpy

import pprint
import numpy


def Matrix():

    N = int(input('请输入数字m：'))
    M = int(input('请输入数字n:'))
    array = numpy.zeros((N, M), dtype=numpy.int16)
    
    # 起始点
    x, y = 0, 0
    res = array[x][y] = 1
    
    while (res < N * M):
        # 改变起始的位置，可以改变旋转，但必须按规律来
        # 上  左-->右
        while (y + 1 < M and not array[x][y + 1]):
            res += 1
            y += 1
            array[x][y] = res
        # 右  上-->下
        while (x + 1 < N and (not array[x + 1][y])):
            res += 1
            x += 1
            array[x][y] = res
        # 下  右--->左
        while (y - 1 >= 0 and not array[x][y - 1]):
            res += 1
            y -= 1
            array[x][y] = res
        #左   下--->上
        while (x - 1 >= 0 and not array[x - 1][y]):
            res += 1
            x -= 1
            array[x][y] = res

    pprint.pprint(array)



if __name__ == '__main__':
    Matrix()