python 修改我的代码 做出如下题目（需要代码，一旦通过所有testcase便采纳）

Question

stri=input('string: ')
cha=input('character: ')
cnt=stri.count(cha)
i=1

for j in stri:
    for i in range(1,cnt+1):
        if j==cha:
            b=stri.replace(j,str(i))
        if i>9:
            break
        
print(b)

Answer 1

stri = str(input('string: '))
cha = str(input('character: '))
cnt = stri.count(cha)

list_str = list(stri)

n = 1
for i in range(len(list_str)):
    if list_str[i] == cha:
        list_str.insert(i+1, str(n))
        list_str.pop(i)
        n += 1
        if n > 9:
            break

res = "".join(list_str)
print(res)