使用函数求特殊a串数列和

【问题描述】给定两个均不超过9的正整数a和n，要求编写函数fn(a,n) 求a+aa+aaa++⋯+aa⋯aa(n个a）之和，fn须返回的是数列和【输入形式】2,3
【输出形式】246
【样例说明】2+22+222=246
【评分标准】

def fn(a,n):
    lst=[]
    for i in range (1,n+1):
        lst.append(str(a)*i)
    lst=list(map(int,lst))
    return sum(lst)
a,n=map(int,input().split(","))
print(fn(a,n))