https://travelebookdownload.blogspot.com/2024/03/python-matrix-manipulation-square-each.html