Python：字符串排形状

某公司的 N位（N>=4）研发同学组织了一次秋游活动，某同学带了个无人机在高空拍照，活动结束时,有人提议排成Y字形来拍一张合照。
用字符串中的每一个字符（不是换行符或结束符’\0’）代表一位老师，输出排好后的队形。要求 Y字除去中心点外，上下半部分等高，按照从左到右，从上到下进行排序。队形中没人的部分用空格占位。
输入数据保证可以排出一个完整的Y字，即长度为 3k+1 （k>=1）
例如: 7个 x ，排成队形为（为了方便说明，这里用‘-’代替空格）：
x—x
-x-x
–x
–x
–x

num = int(input())
arr = input()
k = 1
i = 0   #存放Y头有几行
while k != num :
    k += 3
    i += 1
row = 2*i + 1   #存放一共有几行
k = 0
j = 0
while i>0 :
    for m in range(j):
        print(' ',end='')
    print(arr[k],end='')
    k += 1
    for m in range(2*i-1):
        print(' ',end='')
    print(arr[k])
    k += 1
    i -= 1
    j += 1
temp = j
while k < num :
    for m in range(temp):
        print(' ',end='')
    print(arr[k])
    k += 1