快捷導(dǎo)航

用Python實(shí)現(xiàn)的等差數(shù)列方式

更新時(shí)間：2022年12月16日 16:06:36 作者：ztenv

這篇文章主要介紹了用Python實(shí)現(xiàn)的等差數(shù)列方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

Python實(shí)現(xiàn)等差數(shù)列

def arithemtic_sequence_based_generator(first=0, step=1, sequence_count=10):
    for index in range(0, sequence_count):
        yield first + index * step
        index += 1

if __name__ == "__main__":
    generator = arithemtic_sequence_based_generator(first=0, step=5, sequence_count=10)
    for number in generator:
        print(number, end=" ")

運(yùn)行結(jié)果：

0 5 10 15 20 25 30 35 40 45

python等差數(shù)列生成器

典型的迭代器模式作用很簡(jiǎn)單——遍歷數(shù)據(jù)結(jié)構(gòu)。

不過(guò)，即便不是從集合中獲取元素，而是獲取序列中即時(shí)生成的下一個(gè)值時(shí)，也用得到這種基于方法的標(biāo)準(zhǔn)接口。

例如，內(nèi)置的 range 函數(shù)用于生成有窮整數(shù)等差數(shù)列（Arithmetic Progression，AP）， itertools.count 函數(shù)用于生成無(wú)窮等差數(shù)列。

先來(lái)看等差數(shù)列：

"""
等差數(shù)列實(shí)現(xiàn)。
"""

class ArithmeticProgression:

    def __init__(self, begin, step, end=None):
        self.begin = begin
        self.step = step
        self.end = end   # None -> 無(wú)窮數(shù)列

    def __iter__(self):
        result = type(self.begin + self.step)(self.begin)
        forever = self.end is None
        index = 0
        while forever or result < self.end:
            yield result
            index += 1
            result = self.begin + self.step * index

if __name__ == '__main__':
    ap = ArithmeticProgression(0, 1, 3)
    print(ap)
    # print(list(ap))
    a = iter(ap)
    print(next(a))

這是個(gè)簡(jiǎn)單的示例，說(shuō)明了如何使用生成器函數(shù)實(shí)現(xiàn)特殊的 __iter__ 方法。

然而，如果一個(gè)類只是為了構(gòu)建生成器而去實(shí)現(xiàn) __iter__ 方法，那還不如使用生成器函數(shù)。

畢竟，生成器函數(shù)是制造生成器的工廠。

def aritprog_gen(begin, step, end=None):
    result = type(begin + step)(begin)
    forever = end is None
    index = 0
    while forever or result < end:
        yield result
        index += 1
        result = begin + step * index

使用itertools模塊生成等差數(shù)列

然而，itertools.count 函數(shù)從不停止，因此，如果調(diào)用 list(count())，Python 會(huì)創(chuàng)建一個(gè)特別大的列表，超出可用內(nèi)存，在調(diào)用失敗之前，電腦會(huì)瘋狂地運(yùn)轉(zhuǎn)。

不過(guò)，itertools.takewhile 函數(shù)則不同，它會(huì)生成一個(gè)使用另一個(gè)生成器的生成器，在指定的條件計(jì)算結(jié)果為 False 時(shí)停止。因此，可以把這兩個(gè)函數(shù)結(jié)合在一起使用，編寫下述代碼：

利用 takewhile 和 count 函數(shù)，寫出的代碼流暢而簡(jiǎn)短。

import itertools 


# 注意，下面示例中的 aritprog_gen 不是生成器函數(shù)，
# 因?yàn)槎x體中沒(méi)有 yield 關(guān)鍵字。
# 但是它會(huì)返回一個(gè)生成器，
# 因此它與其他生成器函數(shù)一樣，
# 也是生成器工廠函數(shù)。
def aritprog_two_gen(begin, step, end=None):
    first = type(begin + step)(begin)
    ap_gen = itertools.count(first, step)
    if end is not None:
        ap_gen = itertools.takewhile(lambda n: n < end, ap_gen)
    return ap_gen