Вывод результата в файл

Хочу сделать вывод результатов выполнения программы в файл. Но когда подключаю sys, перестает работать функция. Я не понимаю что нужно сделать что бы все работало как надо, поскольку первый раз работаю с выводом в файл, а времени мало. Подскажите пожалуйста что исправить

import sys
f = open('output.txt','w'); sys.stdout = f 

def matrica():
    r=0
    print("Input first index matrix")
    x = int(input())
    print("Input second index matrix")
    y = int(input())
    matrica=[]
    mm=[]
    #1
    for i in range (x):
        matrica.append([])
        for j in range(y):
            matrica[i].append(int(input("Ведите значение:")))
            r+=1

    #2
    print("Матрица:")
    for u in range(x):
        print(matrica[u])
        mm.append(matrica[u])
    f.write(*mm, sep="\n") 
    #print(*mm, sep="\n")

    #3
    nyli=0

    for i in range (len (matrica)):  
            for j in range (len (matrica[i])):
                if (matrica[i][j]) == 0:
                    nyli += 1
    f.write("Количество рядов с 0 элементом "+nyli)
    #print(nyli)  

    symma=0    
    for i in range (len (matrica)):  
            for j in range (len (matrica[i])):
                if (matrica[i][j]) > 0:
                    symma += matrica[i][j]
    f.write("Суммма положительных чисел "+symma) 
    #print(symma)
    return symma
   
matrica()
f.close()