cplot Lambert series

cplot-lambert-series.py

import cplot
import numpy as np


def lambert_series(z, n=100):
    zn = z.copy()
    s = np.zeros_like(z)
    for _ in range(n):
        s += zn / (1 - zn)
        zn *= z
    return s


plt = cplot.plot(lambert_series, (-1.1, 1.1, 500), (-1.1, 1.1, 500))
plt.savefig("lambert-series.png", bbox_inches="tight")
plt.show()