mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-02-17 06:48:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

rotate_matrix.py: Add type hints for return values (#1023)

Browse Source 
* rotate_matrix.py: Add type hints for return values

@obelisk0114 Your review please?

* Fix typo

* Run the code thru python/black

https://github.com/python/black

* Fix 270 comment

* Simplify with get_data() and test the alternatives

* ) * 3

* Update rotate_matrix.py

* Update rotate_matrix.py
This commit is contained in:
Christian Clauss 2019-07-23 00:07:09 +02:00 committed by GitHub
parent 05e567c2f9
commit b2ed8d443c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 58 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
matrix/rotate_matrix.py
View File

@ -1,99 +1,100 @@
# -*- coding: utf-8 -*-
"""
In this problem, we want to rotate the matrix elements by 90, 180, 270 (counterclockwise)
Discussion in stackoverflow:
https://stackoverflow.com/questions/42519/how-do-you-rotate-a-two-dimensional-array
In this problem, we want to rotate the matrix elements by 90, 180, 270 (counterclockwise)
Discussion in stackoverflow:
https://stackoverflow.com/questions/42519/how-do-you-rotate-a-two-dimensional-array
"""
def rotate_90(matrix: [[]]):
def make_matrix(row_size: int = 4) -> [[int]]:
"""
>>> rotate_90([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
>>> make_matrix()
[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
>>> make_matrix(1)
[[1]]
>>> make_matrix(-2)
[[1, 2], [3, 4]]
>>> make_matrix(3)
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> make_matrix() == make_matrix(4)
True
"""
row_size = abs(row_size) or 4
return [[1 + x + y * row_size for x in range(row_size)] for y in range(row_size)]
def rotate_90(matrix: [[]]) -> [[]]:
"""
>>> rotate_90(make_matrix())
[[4, 8, 12, 16], [3, 7, 11, 15], [2, 6, 10, 14], [1, 5, 9, 13]]
>>> rotate_90(make_matrix()) == transpose(reverse_column(make_matrix()))
True
"""
transpose(matrix)
reverse_row(matrix)
return matrix
def rotate_180(matrix: [[]]):
return reverse_row(transpose(matrix))
# OR.. transpose(reverse_column(matrix))
def rotate_180(matrix: [[]]) -> [[]]:
"""
>>> rotate_180([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
>>> rotate_180(make_matrix())
[[16, 15, 14, 13], [12, 11, 10, 9], [8, 7, 6, 5], [4, 3, 2, 1]]
>>> rotate_180(make_matrix()) == reverse_column(reverse_row(make_matrix()))
True
"""
reverse_column(matrix)
reverse_row(matrix)
"""
OR
reverse_row(matrix)
reverse_column(matrix)
"""
return matrix
def rotate_270(matrix: [[]]):
return reverse_row(reverse_column(matrix))
# OR.. reverse_column(reverse_row(matrix))
def rotate_270(matrix: [[]]) -> [[]]:
"""
>>> rotate_270([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
>>> rotate_270(make_matrix())
[[13, 9, 5, 1], [14, 10, 6, 2], [15, 11, 7, 3], [16, 12, 8, 4]]
>>> rotate_270(make_matrix()) == transpose(reverse_row(make_matrix()))
True
"""
transpose(matrix)
reverse_column(matrix)
"""
OR
reverse_row(matrix)
transpose(matrix)
"""
return matrix
return reverse_column(transpose(matrix))
# OR.. transpose(reverse_row(matrix))
def transpose(matrix: [[]]):
def transpose(matrix: [[]]) -> [[]]:
matrix[:] = [list(x) for x in zip(*matrix)]
return matrix
def reverse_row(matrix: [[]]):
def reverse_row(matrix: [[]]) -> [[]]:
matrix[:] = matrix[::-1]
return matrix
def reverse_column(matrix: [[]]):
def reverse_column(matrix: [[]]) -> [[]]:
matrix[:] = [x[::-1] for x in matrix]
return matrix
def print_matrix(matrix: [[]]):
def print_matrix(matrix: [[]]) -> [[]]:
for i in matrix:
print(*i)
if __name__ == '__main__':
matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
if __name__ == "__main__":
matrix = make_matrix()
print("\norigin:\n")
print_matrix(matrix)
rotate_90(matrix)
print("\nrotate 90 counterclockwise:\n")
print_matrix(matrix)
print_matrix(rotate_90(matrix))
matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
matrix = make_matrix()
print("\norigin:\n")
print_matrix(matrix)
rotate_180(matrix)
print("\nrotate 180:\n")
print_matrix(matrix)
print_matrix(rotate_180(matrix))
matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
matrix = make_matrix()
print("\norigin:\n")
print_matrix(matrix)
rotate_270(matrix)
print("\nrotate 270 counterclockwise:\n")
print_matrix(matrix)
print_matrix(rotate_270(matrix))