Python生成器的使用方法和示例代碼

更新時間：2019年03月04日 15:32:56 作者：閼男秀

今天小編就為大家分享一篇關(guān)于Python生成器的使用方法和示例代碼，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧

本文是《Effect Python 編寫高質(zhì)量Python代碼的59個有效方法》的學習筆記。主要記錄生成器的使用方法和示例代碼。

返回隊列的函數(shù)

如果函數(shù)要產(chǎn)生一系列結(jié)果，那么最簡單的做法就是把這些結(jié)構(gòu)都放在一份列表里，然后將其返回給調(diào)用者。

def index_words(text):
  """用append方法將這些此的首字母索引添加到result列表中，并在函數(shù)結(jié)束時將其返回給調(diào)用者。"""
  result = []
  if text:
    result.append(0)
  for index, letter in enumerate(text):
    if letter == ' ':
      result.append(index+1)
  return result

輸入一些測試值，驗證該函數(shù)能正常運行：

>address = 'Four score and seven years ago...'
result = index_words(address)
print(result[:3])

打印

[0, 5, 11]

生成器函數(shù)

這個函數(shù)改用生成器（generator）來寫會更好。生成器是使用yield表達式的函數(shù)。調(diào)用生成器函數(shù)時，它并不會真的運行，而是會返回迭代器。每次在這個迭代器上面調(diào)用內(nèi)置的next函數(shù)時，迭代器會把生成器推進到下一個yield表達式那里。生成器傳給yield的每一個值，都會由迭代器返回給調(diào)用者。

def index_words_iter(text):
  if text:
    yield 0
  for index, letter in enumerate(text):
    if letter == ' ':
      yield index + 1

result = list(index_words_iter(address))

注意：生成器函數(shù)返回的迭代器是有狀態(tài)的，調(diào)用者不應該反復使用。

由于迭代器只能產(chǎn)生一輪結(jié)果。在拋出過StopIteration異常的迭代器或生成器上面繼續(xù)迭代第二輪，是不會有結(jié)果的。為解決此問題，我們可以明確地使用該迭代器制作一份列表，將它的全部內(nèi)容都遍歷一次，并賦值到這份列表里，然后就可以在復制出來的數(shù)據(jù)列表上面多次迭代了。

為類實現(xiàn)生成器

下面是一個可以迭代的容器類，用來從文件中讀取每行數(shù)據(jù)。

class ReadFileLines(object):
  """
  可以迭代的容器類，從文件中獲取數(shù)據(jù)
  """
  def __init__(self, path):
    self.path = path
  def __iter__(self):
    with open(self.path) as f:
      for line in f:
        yield line

多次迭代

如果想多次迭代生成器的數(shù)據(jù)，可以使用下面的函數(shù)。該函數(shù)會逐步拷貝生成器的所有數(shù)據(jù)，然后返回一個隊列數(shù)據(jù)。

def normalize_defensive(datas):
  """
  從生成器返回一份可以多次迭代的數(shù)據(jù)
  :param datas:容器 
  :return: result:隊列(list)
  """
  # 確保調(diào)用者傳進來的參數(shù)，并不是迭代器對象本身
  if iter(datas) is iter(datas):
    raise TypeError('Must supply a container')
  # TODO
  result = []
  for data in datas:
    # TODO
    result.append(data)
  return result

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對腳本之家的支持。如果你想了解更多相關(guān)內(nèi)容請查看下面相關(guān)鏈接

您可能感興趣的文章: