example hilbert curves for khalid

khalid_hilbert.py

import numpy as np
import time
from hilbertcurve.hilbertcurve import HilbertCurve


N = 40
num_vecs = 100_000


for p in [1, 2, 3, 4]:

    hilbert_curve = HilbertCurve(p, N)
    print("N={}, p={}".format(N, p))
    print("coordinate limits in any dimension: ({}, {})".format(0, hilbert_curve.max_x))
    print("max distance along curve: {}".format(hilbert_curve.max_h))
    print("calculating distance along curve for {} {}-dimensional coordinates".format(num_vecs, N))

    t0 = time.time()
    coords = np.random.randint(0, hilbert_curve.max_x + 1, size=(num_vecs, N))
    dists = []
    for coord in coords:
        dist = hilbert_curve.distance_from_coordinates(coord)
        dists.append(dist)
    t1 = time.time()
    print("dt = {:.3f}s".format(t1-t0))
    print()