2次元配列を回転したい

2次元配列を回転したい

# 二次元配列を回転したい

頭が混乱してしまうのでまとめる．

# 反時計回りに 90 度回転

回転後の行列を左上から右下へ埋めていくイメージ
つまり回転前の行列の右上から左下へ拾っていく
僕の頭的には，回転後の行列の座標の軌跡を正として回転前の位置を参照する方が書きやすいみたい．

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
def rotate_counterclockwise(table):
    ROW = len(table[0]) # 回転後の行数は回転前の列数
    COL = len(table)    # 回転後の列数は回転前の行数

    rotated = [[0 for _ in range(COL)] for _ in range(ROW)]

    for r in range(ROW):
        for c in range(COL):
            rotated[r][c] = table[c][ROW - r - 1]

    return rotated

# 時計回りに 90 度回転

回転後の行列を左上から右下へ埋めていくイメージ
つまり回転前の行列の左下から右上へ拾っていく

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
def rotate_clockwise(table):
    ROW = len(table[0]) # 回転後の行数は回転前の列数
    COL = len(table)    # 回転後の列数は回転前の行数

    rotated = [[0 for _ in range(COL)] for _ in range(ROW)]

    for r in range(ROW):
        for c in range(COL):
            rotated[r][c] = table[COL - c - 1][r]

    return rotated

# かっこいい書き方

zipを使うと左 90 度回転はかっこよくかける
zipで回転後の行列を行ごとに下から集めたものが取れるので，最後にそれを逆順にする
ref: https://stackoverflow.com/questions/8421337/rotating-a-two-dimensional-array-in-python

1
2
def rotate_counterclockwise(table):
    return [list(row) for row in zip(*table)][::-1]

豆知識：左 90 度回転 x 3 == 右 90 度回転