Python：如何在不同的函数中使用写入/导出到文件？

Question

我想在Python中编写一个导出到文件函数。这里有一个片段：

#!/usr/bin/env python 
def function_1(x): 
    for i in range(len(x)): 
     ip = packet[i] 
     mac = mac[i] 
     print '{}, {}'.format(ip, mac) 

def export() 
    f=open("data.txt", "w") 
    f.write() #f.write(function_1(x))?? 
    f.close() 

if __name__ == "__main__": 
    function_1(10) 
    export() 

我可以把它在相同的功能function_1做写入一个文件，但我会怎么做它在另一个功能？我需要返回吗？

Answer 1

是的，f.write需要一个字符串输入。最简单的方法是直接在函数中创建字符串并将其返回：

def function_1(x): 
    ret_val = "" 
    for i in range(len(x)): 
     ip = packet[i] 
     mac = mac[i] 
     ret_val += '{}, {}\n'.format(ip, mac) 
    return ret_val