如何在不同的行和列中获取给定的字符串？

Question

我正在尝试解决HackerRank的问题之一，我已经解决了一半，但是在其中一个测试用例中出现了错误。 Link的问题

输入： haveaniceday

我的代码输出： hae and via ecy（正确）

输入： feedthedog

输出：

  

字符串索引超出范围

代码：

import math
def fun(a,c,f):
    l=[]
    if c*f>=len(a):
        for i in range(0,f):
            z=a[i*c:c*(i+1)]
            l.append(z)
        return l
    else:
        f=c
        for i in range(0,f):
            z=a[i*c:c*(i+1)]
            l.append(z)
        return l 
n="feedthedog"
l=len(n)
z=l**(1/2)
f=math.floor(z)
c=math.ceil(z)
ans=fun(n,c,f)
print(ans)
ll=[]
for i in range(0,c):
    x=""
    for j in range(0,f):
        x+=ans[j][i]
    ll.append(x)
for i in ll:
    print(i,end=" ")