Vortex

vortex.py

import numpy as np
from matplotlib import collections as mc, pyplot as plt

Point = tuple[float, float]
Line = tuple[Point, Point]

def get_lines(multiplier: int, modulus: int) -> list[Line]:
    start = np.arange(1, modulus)
    end = start * multiplier % modulus
    angle = 2 * np.pi / modulus
    points = lambda x: np.column_stack((np.cos(x), np.sin(x)))
    return list(zip(points(start * angle), points(end * angle)))

def diagram(multiplier: int, modulus: int, linewidth: float = 0.5):
    lines = get_lines(multiplier, modulus)
    _, ax = plt.subplots()
    ax.add_collection(mc.LineCollection(lines, linewidths=linewidth))
    plt.axis("square")
    plt.show()

if __name__ == "__main__":
    diagram(240, 7417, 0.01)