怎么样用python把输出结果放进txt里面呢

怎么样新建一个txt，把这个菱形输出写入文件里呢，然后保存下来

def fun(n, line, inx, flag=1):
    if line == n + 1 or line == 0: return inx

    for ii in range(n - line):
       print(end=' ')
       f.write(' ')

    for i in range(line):
        print(inx % 10, end=' ')
        f.write(str(inx % 10)+' ')
        inx += 1
    print()
    f.write('\n')
    line += flag
    t = fun(n, line, inx, flag)
    return t

f = open('test.txt','w')
n = 6
t = fun(n, 1, 0)
fun(n, n - 1, t, -1)
f.close()