"""
Quantile-Quantile Plot
----------------------
A quantile-quantile plot comparing input data to theoretical distributions.
"""
# category: scatter plots

import altair as alt
from vega_datasets import data

source = data.normal_2d.url

base = alt.Chart(source).transform_quantile(
    'u',
    step=0.01,
    as_ = ['p', 'v']
).transform_calculate(
    uniform = 'quantileUniform(datum.p)',
    normal = 'quantileNormal(datum.p)'
).mark_point().encode(
    alt.Y('v:Q')
)

base.encode(x='uniform:Q') | base.encode(x='normal:Q')

add qq example