2020华为春招--N阶方阵旋转(python)

题目描述：输入一个N阶方阵(0<N<10),输出此方阵顺时针旋转M(0<=M<=10000)次后的方阵
题目示例：三阶方阵，围绕方阵中心顺时针旋转

输入描述：
(1) 第一行输入一个正整数N (0<N<10)
(2) 接下来输入方阵的数据，输入N行，每行N个整数用空格分开
(3) 最后输入一个正整数M (0<=M<=10000)
说明：不用考虑异常输入，所有输入都是正常的
输出描述：
(1) 输出旋转后的数据，共计N行，每行N个整数，用空格分开。
输入示例：
3
1 2 3
4 5 6
7 8 9
2
输出示例：
9 8 7
6 5 4
3 2 1

import sys
import math

n = int(sys.stdin.readline().strip())
array = []
for i in range(n):
        # 读取每一行
        line = sys.stdin.readline().strip()
        # 把每一行的数字分隔后转化成int列表
        rows = list(map(int,line.split()))
        array.append(rows)

def rotation_matrix(array,n):
    # 二维数组存在浅复制的问题，需要采用二维列表推导式
    arr2 =  [([0]*n) for i in range(n)]
    # math.ceil 向上取整，目的是为了将矩阵的中心位置赋值
    for i in range(math.ceil(n/2)):
        for j in range(i,n-i):
            # 第i列转到第i行
            arr2[i][j] = array[n-1-j][i] 
            # 第i行转到第n-i列
            arr2[j][n-i-1] = array[i][j]
            # 第n-i列转到n-i行
            arr2[n-i-1][j] = array[n-j-1][n-i-1]
            # 第n-i行转到第i列
            arr2[j][i] = array[n-i-1][j]
    return arr2
# 旋转次数
m = int(sys.stdin.readline().strip())
for i in range(m):
    array = rotation_matrix(array,n)
for i in range(n):
    for j in range(n):
        print(array[i][j],end=' ')
    print('')