python 卡号添加空格

#给卡号添加空格，如卡号’6212261702018115014’,处理后，返回’621 2261 7020 1811 5014’
def kg(s):
    s1=s[:3]+' '
    s2=s[3:]
    #加空格
    a=len(s2);n=1
    while 4*n<=a:
        s2=s2[:5*n-1]+' '+s2[5*n-1:]
        n+=1
    return(s1+s2)
s='6212261702018115014'
print(kg(s))