函数使用print进行迭代但不返回

Question

Python新手在这里,运行2.7.

我正在尝试创建一个程序,该程序使用函数生成文本,然后将函数生成的文本输出到文件.

当只在powershell中打印函数时(如:http://codepad.org/KftHaO6x),它会迭代,因为我希望它:

def writecode (q, a, b, c):
    while b < q:
        b = b + 1
        print "v%d_%d_%d = pairwise (caps[%d],sals[%d],poss[%d],poss[%d],poss[%d],pos_range)" %(a,b,c,a,a,a,b,c)
        print "votes%d_%d.append(v%d_%d_%d)" % (b,c,a,b,c,)
        print "v%d_%d_%d = pairwise (caps[%d],sals[%d],poss[%d],poss[%d],poss[%d],pos_range)" %(a,c,b,a,a,a,c,b)
        print "votes%d_%d.append(v%d_%d_%d)" % (c,b,a,c,b)

writecode (5,1,0,4)

当尝试将函数输出到文件中时(例如:http://codepad.org/8GJpp9QY),它只给出1个值,即不迭代:

def writecode (q, a, b, c):
    while b < q:
        b = b + 1
        data_to_write = "v%d_%d_%d = pairwise (caps[%d],sals[%d],poss[%d],poss[%d],poss[%d],pos_range)" %(a,b,c,a,a,a,b,c)
        data_to_write_two = "votes%d_%d.append(v%d_%d_%d)" % (b,c,a,b,c,)
        data_to_write_three = "v%d_%d_%d = pairwise (caps[%d],sals[%d],poss[%d],poss[%d],poss[%d],pos_range)" %(a,c,b,a,a,a,c,b)
        data_to_write_four = "votes%d_%d.append(v%d_%d_%d)" % (c,b,a,c,b)
        return data_to_write
        return data_to_write_two
        return data_to_write_three
        return data_to_write_four

x = writecode (5,1,0,4)

out_file = open("code.txt", "a")
out_file.write(x)
out_file.close()

为什么会这样,我怎样才能使输出函数迭代(就像打印一样)？

Answer 1

你的问题是你只能return从一个功能:

def test():
    return 1
    return 2

x = test()
# x is now 1

您将要使用yield或返回值元组:

def this_will_work():
    yield 1
    yield 2

x = list(this_will_work()) # See below for why we are using list
# x is now [1, 2]

def so_will_this:
    return 1, 2

x = so_will_this()
# x is now (1, 2)

使用yield将您的函数转换为generator可以迭代的函数- 如果您只对所有最终值感兴趣,可以使用list这些函数将生成器转换为生成器生成的所有值的列表.或者你可以使用它循环它for ... in.

Answer 2

在您用来写入文件的版本中，函数在循环return的第一次迭代后返回（通过第一个语句）while。根据你所拥有的，你可能想要这样的东西：

def writecode (q, a, b, c):
    results = []
    while b < q:
        b = b + 1
        results.append("v%d_%d_%d = pairwise (caps[%d],sals[%d],poss[%d],poss[%d],poss[%d],pos_range)" % (a,b,c,a,a,a,b,c))
        results.append("votes%d_%d.append(v%d_%d_%d)" % (b,c,a,b,c,))
        results.append("v%d_%d_%d = pairwise (caps[%d],sals[%d],poss[%d],poss[%d],poss[%d],pos_range)" % (a,c,b,a,a,a,c,b))
        results.append("votes%d_%d.append(v%d_%d_%d)" % (c,b,a,c,b))
        results.append("")
    return "\n".join(results)

x = writecode (5,1,0,4)

out_file = open("code.txt", "a")
out_file.write(x)
out_file.close()

其工作原理是将每一行输出累积到一个列表中，然后返回单个字符串，其中所有结果都与换行符和尾随换行符连接在一起。