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

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


class ArithmeticProgression:

  def __init__(self, begin, step, end=None):
    self.begin = begin
    self.step = step
    self.end = end  # None -> 無窮數(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))

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

import itertools 


# 注意，下面示例中的 aritprog_gen 不是生成器函數(shù)，
# 因為定義體中沒有 yield 關鍵字。
# 但是它會返回一個生成器，
# 因此它與其他生成器函數(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