Как повернуть массив на 90 градусов python

Аватар пользователя Игорь Гахов
Игорь Гахов
18 октября 2022

Проще всего это будет сделать с помощью Numpy:

import numpy as np

l = [[0, 0, 0, 0], [0, 0, 0, 1], [0, 0, 0, 1], [0, 0, 0, 1]]

a = np.array(l)

# a = array([[0, 0, 0, 0],
#            [0, 0, 0, 1],
#            [0, 0, 0, 1],
#            [0, 0, 0, 1]], dtype=int64)

# поворот:

np.rot90(a)

# a = array([[0, 1, 1, 1],
#            [0, 0, 0, 0],
#            [0, 0, 0, 0],
#            [0, 0, 0, 0]], dtype=int64)

# можно несколько раз повернуть:

np.rot90(a, k=2)

# a = array([[1, 0, 0, 0],
#            [1, 0, 0, 0],
#            [1, 0, 0, 0],
#            [0, 0, 0, 0]], dtype=int64)

# "вернуть" к обычному списку списков:

np.rot90(a).tolist()
# => [[0, 1, 1, 1], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

Соответственно для поворота по часовой стрелке используйте np.rot90(a, k=3) (270 градусов).

Если не использовать Numpy, то для поворота по часовой стрелке есть такой способ:

def transposed(matrix):
    return [[*col] for col in zip(*matrix)]

def rot90:
    return list(map(reversed, transposed(matrix)))

# или в одно действие

def rot90(matrix):
    return [list(reversed(col)) for col in zip(*matrix)]

... а для поворота против часовой:

def rot90(matrix):
    return [list(row) for row in list(zip(*matrix))[::-1]]
0 0
Познакомьтесь с основами Python бесплатно