approximate_pi.py

def divide(n, d):
  while True:
    yield n // d
    n = (n % d) * 10

def next_n(n, g):
  return [next(g) for _ in range(n)]

def input(number):
  pi = divide(355, 113)
  return next_n(number, pi)

print(input(number=3))
print(input(number=10))