快捷導(dǎo)航

python numpy生成等差數(shù)列、等比數(shù)列的實(shí)例

更新時(shí)間：2020年02月25日 13:13:39 作者：i徒然

今天小編就為大家分享一篇python numpy生成等差數(shù)列、等比數(shù)列的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

如下所示：

import numpy as np
# 等差數(shù)列
print(np.linspace(0.1, 1, 10, endpoint=True))
print(np.arange(0.1, 1.1, 0.1))
"""總結(jié)：
  arange 側(cè)重點(diǎn)在于增量，不管產(chǎn)生多少個(gè)數(shù)
  linspace 側(cè)重于num, 即要產(chǎn)生多少個(gè)元素，不在乎增量
"""
 
# 等比數(shù)列
np.logspace(1, 4, 4, endpoint=True, base=2) # 2**1---2**4

補(bǔ)充拓展：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é)合在一起使用，編寫(xiě)下述代碼：

利用 takewhile 和 count 函數(shù)，寫(xiě)出的代碼流暢而簡(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

以上這篇python numpy生成等差數(shù)列、等比數(shù)列的實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: