Как написать return, если результатом функции есть результат цикла?

Не нужно вызывать print. Просто вызовите функцию тогда None не будет.