python将二维数组逆时针旋转45°

#二维数组逆时针旋转45°
def rotate(array):
    if array==None:
        return 0
    length=len(array)
    #右上角输出
    i=length-1
    while i>0: #外层循环用于控制行数，如n*n矩阵右上角输出有n-1行
        row=0
        col=i
        while col<length:#内层循环用于输出元素，如n=4时，依次输出[0,3];[0,2],[1,3];[0,1],[1,2],[2,3]
            print(array[row][col],end=' ')
            row+=1
            col+=1
        print('\n')
        i-=1      
        
    #左下角输出
    i=0
    while i<length:
        row=i
        col=0
        while row<length:
            print(array[row][col],end=' ')
            row+=1
            col+=1
        print('\n')
        i+=1
if __name__ == '__main__':
    array=[[1,2,3],[4,5,6],[7,8,9]]
    '''1 2 3
       4 5 6
       7 8 9'''
    rotate(array)
'''
旋转结果为:
3 
2 6 
1 5 9 
4 8 
7'''