https://codeworld19.blogspot.com/2020/11/transpose-and-flatten-in-python-hacker.html