古风排版 python

7-9 古风排版 (15 分)

中国的古人写文字，是从右向左竖向排版的。本题就请你编写程序，把一段文字按古风排版。

输入格式：
输入在第一行给出一个正整数N（<100），是每一列的字符数。第二行给出一个长度不超过1000的非空字符串，以回车结束。

输出格式：
按古风格式排版给定的字符串，每列N个字符（除了最后一列可能不足N个）。

输入样例：

4
This is a test case

输出样例：

asa T
st ih
e tsi
 ce s

代码如下：

num = int(input())
string = input()
while len(string)%num != 0:
    string += ' '
sum_number = len(string)//num
mylist = list()
for i in range(sum_number):
    myslice = string[i*num:(i+1)*num]
    mylist.append(myslice)

mylist = list(reversed(mylist))

result_list = list(zip(*mylist))

for i in result_list:
    for index,j in enumerate(i):
        if index == len(i)-1:
            print(j)
        else:
            print(j,end='')