生成器函數學習筆記

• Fluent Python真的是一本經典之作。
• • 演示 itertools.accumulate 函數的幾個用法
  • 演示用于映射的生成器函數
  • itertools.product 生成器函數

Fluent Python真的是一本經典之作。

演示 itertools.accumulate 函數的幾個用法

>>> sample = [5, 4, 2, 8, 7, 6, 3, 0, 9, 1] >>> import itertools >>> list(itertools.accumulate(sample)) # 計算總和[5, 9, 11, 19, 26, 32, 35, 35, 44, 45] >>> list(itertools.accumulate(sample, min)) # 計算最小值[5, 4, 2, 2, 2, 2, 2, 0, 0, 0] >>> list(itertools.accumulate(sample, max)) # 計算最大值[5, 5, 5, 8, 8, 8, 8, 8, 9, 9] >>> import operator >>> list(itertools.accumulate(sample, operator.mul)) # 計算乘積[5, 20, 40, 320, 2240, 13440, 40320, 0, 0, 0] >>> list(itertools.accumulate(range(1, 11), operator.mul)) [1, 2, 6, 24, 120, 720, 5040, 40320, 362880, 3628800] # 從 1! 到 10!，計算各個數的階乘

演示用于映射的生成器函數

>>> list(enumerate('albatroz', 1)) # 從 1 開始，為單詞中的字母編號。[(1, 'a'), (2, 'l'), (3, 'b'), (4, 'a'), (5, 't'), (6, 'r'), (7, 'o'), (8, 'z')] >>> import operator >>> list(map(operator.mul, range(11), range(11))) # 從 0 到 10，計算各個整數的平方。[0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100] >>> list(map(operator.mul, range(11), [2, 4, 8])) # 計算兩個可迭代對象中對應位置上的兩個元素之積，元素最少的那 個可迭代對象到頭后就停止。[0, 4, 16] >>> list(map(lambda a, b: (a, b), range(11), [2, 4, 8])) # 作用等同于內置的 zip 函數。[(0, 2), (1, 4), (2, 8)] >>> import itertools >>> list(itertools.starmap(operator.mul, enumerate('albatroz', 1))) # 從 1 開始，根據字母所在的位置，把字母重復相應的次數。['a', 'll', 'bbb', 'aaaa', 'ttttt', 'rrrrrr', 'ooooooo', 'zzzzzzzz'] >>> sample = [5, 4, 2, 8, 7, 6, 3, 0, 9, 1] >>> list(itertools.starmap(lambda a, b: b/a, enumerate(itertools.accumulate(sample), 1))) # 計算平均值。[5.0, 4.5, 3.6666666666666665, 4.75, 5.2, 5.333333333333333, 5.0, 4.375, 4.888888888888889, 4.5]

itertools.product 生成器函數

>>> list(itertools.product('ABC', range(2))) # 三個字符的字符串與兩個整數的值域得到的笛卡兒積是六個元組 （因為 3 * 2 等于 6）。[('A', 0), ('A', 1), ('B', 0), ('B', 1), ('C', 0), ('C', 1)] >>> suits = 'spades hearts diamonds clubs'.split() >>> list(itertools.product('AK', suits)) # 兩張牌（'AK'）與四種花色得到的笛卡兒積是八個元組。[('A', 'spades'), ('A', 'hearts'), ('A', 'diamonds'), ('A', 'clubs'), ('K', 'spades'), ('K', 'hearts'), ('K', 'diamonds'), ('K', 'clubs')] >>> list(itertools.product('ABC')) # 如果傳入一個可迭代的對象，product 函數產出的是一系列只有一 個元素的元組，不是特別有用。[('A',), ('B',), ('C',)] >>> list(itertools.product('ABC', repeat=2)) # repeat=N 關鍵字參數告訴 product 函數重復 N 次處理輸入的各個 可迭代對象。[('A', 'A'), ('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'B'), ('B', 'C'), ('C', 'A'), ('C', 'B'), ('C', 'C')] >>> list(itertools.product(range(2), repeat=3)) [(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)] >>> rows = itertools.product('AB', range(2), repeat=2) >>> for row in rows: print(row)... ('A', 0, 'A', 0) ('A', 0, 'A', 1) ('A', 0, 'B', 0) ('A', 0, 'B', 1) ('A', 1, 'A', 0)('A', 1, 'A', 1) ('A', 1, 'B', 0) ('A', 1, 'B', 1) ('B', 0, 'A', 0) ('B', 0, 'A', 1) ('B', 0, 'B', 0) ('B', 0, 'B', 1)('B', 1, 'A', 0) ('B', 1, 'A', 1) ('B', 1, 'B', 0)('B', 1, 'B', 1)

總結

以上是生活随笔為你收集整理的生成器函数学习笔记的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。