2012-08-01 79 views
5

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() 

这是为什么,我怎么可以使输出功能迭代(如将其与打印一样)?

+5

的'return'声明** **终止函数(如在大多数(全部)的语言吗?):http://docs.python.org/reference/simple_stmts.html#the-return-statement – 2012-08-01 20:17:14

+3

感谢Hamish将实际的代码放在这里的问题中...... – 2012-08-01 20:18:34

+2

回过头来,你似乎在使用代码生成。你为什么不直接执行代码?或者说一些关于你更大的任务? – 2012-08-01 20:23:27

回答

2

在您用来写入文件的版本中,该函数在while循环的第一次迭代之后返回(通过第一个return语句)。根据你所拥有的,你可能想是这样的:

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() 

其中通过累加输出的每一行到一个列表中工作,然后返回一个字符串与所有的结果与新行和结尾的新行连接在一起。

+0

谢谢,您提供的代码完全符合我需要的内容。非常感激! – user1569317 2012-08-01 20:47:07

+0

@ user1569317:不客气。 FWIW可能有更简单的方法来完成你正在做的事情,或者至少让它更容易阅读。我所做的只是向你展示一种相对简单的方式来获得你的工作。 – martineau 2012-08-02 02:03:07

1

return立即存在的功能:

return离开当前函数调用与表达式列表(或None)作为返回值。

你需要为你去(或者,更好的性能,创造价值的listjoin末)建立的字符串。

5

你的问题是,你只能一次从一个函数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对其进行循环。

0

略有变化;我已经使用带有命名字段的新式字符串格式,这应该更易于阅读。

请注意,您正在生成一些重复的块(在输出中b == c会出现两次的每种情况)。

from textwrap import dedent 

codeblock = dedent("""\ 
    v{a}_{b}_{c} = pairwise (caps[{a}],sals[{a}],poss[{a}],poss[{b}],poss[{c}],pos_range) 
    votes{b}_{c}.append(v{a}_{b}_{c}) 
    v{a}_{c}_{b} = pairwise (caps[{a}],sals[{a}],poss[{a}],poss[{c}],poss[{b}],pos_range) 
    votes{c}_{b}.append(v{a}_{c}_{b}) 
""") 

def get_code (q, a, b, c): 
    return ''.join(codeblock.format(a=a, b=b, c=c) for b in xrange(b, q)) 

def main(): 
    with open('code.txt', 'a') as outf: 
     outf.write(get_code(5,1,0,4)) 

if __name__=="__main__": 
    main()