python - 对于将输出写入 .txt 文件的函数，return 与 print

Question

我正在尝试将以下代码的输出写入 .txt 文件。

import os 

def add(num_1, num_2):
    result_1 = num_1 + num_2
    result_2 = 10 + result_1
    return("Hello, World!")
    return(str(result_1))
    return(str(result_2)) 

output = add(1,7)
save_path = 'file_path'
file_name = os.path.join(save_path, "test.txt")
file = open(file_name,'w')
file.write(output)
file.close()

当我打开 .txt 文件时，唯一写的是“Hello, World!”

但是，如果我将返回更改为打印并在 iPython 控制台而不是 .txt 文件中输出，我会得到以下结果。

def add_2(num_1, num_2):
    result_1 = num_1 + num_2
    result_2 = 10 + result_1
    print("Hello, World!")
    print(str(result_1))
    print(str(result_2)) 

add_2(3,7)

我得到以下输出: Hello World !1020

如何将函数的所有返回值放入 .txt 文件中？

谢谢。

Answer 1

import os 

def add(num_1, num_2):
    result_1 = num_1 + num_2
    result_2 = 10 + result_1
    return("Hello, World!")
    return(str(result_1))
    return(str(result_2)) 

output = add(1,7)
save_path = 'file_path'
file_name = os.path.join(save_path, "test.txt")
file = open(file_name,'w')
file.write(output)
file.close()

当你运行上面的代码时，一旦它遇到 add 函数中的第一个 return，它就会停止运行，所以 return(str(result_1)) 和 return(str(result_2)) 从未被处理过。

输出相当于:

    return("Hello, World!")
    return(str(result_1))
    return(str(result_2)) 

你需要这样写:

    return "Hello, World!" + '\n' + str(result_1) + '\n' + str(result_2)