如何將函數的結果存儲在列表中？

Question

我有以下功能：

def second(n): 
    z =0 
    while z < n: 
     z = z +1 
     for i in range(n): 
      print z 

輸出將是第二（3）：

1 
1 
1 
2 
2 
2 
3 
3 
3 

我已經試過幾件事情，但似乎無法存儲「結果」在列表中供進一步使用。有沒有人知道我應該嘗試什麼？

Answer 1

打印不會保存結果 - 它只是將它們（一次一個）發送到標準輸出（標準輸出）以顯示在屏幕上。要獲得結果副本，我虛心地建議您使用yield關鍵字將其變成generator function。例如：

>>> def second(n): 
...  z =0 
...  while z < n: 
...   z = z +1 
...   for i in range(n): 
...    yield z 
... 
>>> print list(second(3)) 
[1, 1, 1, 2, 2, 2, 3, 3, 3] 

在上面的代碼list()擴展到發電機可以分配給一個變量來保存結果的列表。如果您想保存second(3)的所有結果列表，您可以選擇使用results = list(second(3))而不是打印它。

Answer 2

你可以利用itertools這裏返回一個可迭代的你，無論是環比的是，或顯式轉換爲list，如：

from itertools import chain, repeat 

def mysecond(n): 
    return chain.from_iterable(repeat(i, n) for i in xrange(1, n)) 

print list(mysecond(3)) 

Answer 3

這裏是一個生成器表達式例如：

list(j+1 for i in xrange(n) for j in [i]*n)