linear_complex.py

# -*- coding: utf-8 -*-

import cmath
import math

import matplotlib.pyplot as plt


def linear(a, z, n=100):
    yield z
    for _ in range(n):
        z = a * z
        yield z
        

def main():
    length = 0.9
    angle = 1
    a = complex(math.sin(angle), math.cos(angle)) * length
    seed = complex(1, 0)

    sequence = [c for c in linear(a, seed, 100)]
    real = [c.real for c in sequence]
    imag = [c.imag for c in sequence]
    plt.plot(real, imag)
    plt.show()

    
if __name__ == '__main__':
    main()