04_perfplot2.py

def fixed_query(size):
    """
    Returns a random array of given size and
    a fixed number of random query points.
    """
    array = random_array(size)
    query_points = random_array(1e4)
    return (array, query_points)


plt.figure(figsize=(10, 10))
plt.title('Quantitative Comparison, Fixed Query Size')
perfplot.show(
    setup=fixed_query,
    kernels=[kdtree, multiple_queries_index, multiple_queries],
    n_range=[2**k for k in range(2, 22)],
    logx=True,
    logy=True,
    equality_check=False,
    xlabel='len(df)')
plt.show()