1. 程式人生 > >Python之2維list轉置、旋轉及其簡單應用

Python之2維list轉置、旋轉及其簡單應用

Python之2維list轉置、旋轉及其簡單應用

 給一個矩陣,順時針旋轉順序輸出其元素,例如:

對於矩陣:

[ 1, 2, 3 ]

[ 4, 5, 6 ]

[ 7, 8, 9 ]

 輸出為:

1,2,3,6,9,8,7,4,5

 

複製程式碼

def transpose(matrix):
    return zip(*matrix)

def rotate(matrix):
    return zip(*matrix)[::-1]

def rotatePrint(matrix):
    import copy
    matrix = copy.deepcopy(matrix)
    return matrix and list(matrix.pop(0)) + rotatePrint(rotate(matrix))


######### Test #############

def printMatrix(matrix):
    for row in  matrix:
        print ' '.join( str(i) for i in row)

matrix = [
[1,2,3],
[4,5,6],
[7,8,9]]

print 'original:'
printMatrix(matrix)

print 'rotate print:',rotatePrint(matrix)

print 'transpose:'
printMatrix( transpose(matrix) )

print 'rotate:'
printMatrix(  rotate(matrix))

複製程式碼

輸出為: