Interactive ggplot#

The ggplot API allows us to build different types of of graphics

To make our ggplot interactive, we can use interact API from Jupyter Widgets

Interact autogenerates UI controls for function arguments, and then calls the function with those arguments when you manipulate the controls interactively.

To use interact, you need to define:

  1. Widgeets to be controlled

  2. The plot function includes ggplot with dynamic argument as

  3. Invoke interact() API

Let’s see examples below!

Examples#

Setup#

%load_ext sql
%sql duckdb://

from sql.ggplot import ggplot, aes, geom_histogram, facet_wrap
import ipywidgets as widgets
from ipywidgets import interact

from pathlib import Path
from urllib.request import urlretrieve

url = "https://d37ci6vzurychx.cloudfront.net/trip-data/yellow_tripdata_2021-01.parquet"

if not Path("yellow_tripdata_2021-01.parquet").is_file():
    urlretrieve(url, "yellow_tripdata_2021-01.parquet")

Basic Usage (with Dropdown and Slider widgets)#

dropdown = widgets.Dropdown(
    options=["red", "blue", "green", "magenta"],
    value="red",
    description="Color:",
    disabled=False,
)
b = widgets.IntSlider(
    value=5,
    min=1,
    max=10,
    step=1,
    description="Bin:",
    orientation="horizontal",
)

def plot_fct(color, b):
    (
        ggplot(
            table="yellow_tripdata_2021-01.parquet",
            mapping=aes(x="trip_distance", fill=color),
        )
        + geom_histogram(bins=b)
    )


interact(plot_fct, color=dropdown, b=b)

<function __main__.plot_fct(color, b)>

Categorical histogram (with Select widget)#

Prepare dataset#

We also use ggplot2 diamonds to demonstrate

from pathlib import Path
from urllib.request import urlretrieve

if not Path("diamonds.csv").is_file():
    urlretrieve(
        "https://raw.githubusercontent.com/tidyverse/ggplot2/main/data-raw/diamonds.csv",  # noqa
        "diamonds.csv",
    )

%%sql
CREATE TABLE diamonds AS SELECT * FROM diamonds.csv
Running query in 'duckdb://'
Count
53940

Multiple Columns#

columns = widgets.SelectMultiple(
    options=["cut", "color"], value=["cut"], description="Columns", disabled=False
)

def plot(columns):
    (ggplot("diamonds", aes(x=columns)) + geom_histogram())


interact(plot, columns=columns)

<function __main__.plot(columns)>

cmap = widgets.Dropdown(
    options=["viridis", "plasma", "inferno", "magma", "cividis"],
    value="plasma",
    description="Colormaps:",
    disabled=False,
)

def plot(cmap):
    (
        ggplot("diamonds", aes(x="price"))
        + geom_histogram(bins=10, fill="cut", cmap=cmap)
    )


interact(plot, cmap=cmap)

<function __main__.plot(cmap)>

Facet wrap (Complete Example)#

b = widgets.IntSlider(
    value=5,
    min=1,
    max=10,
    step=1,
    description="Bin:",
    orientation="horizontal",
)
cmap = widgets.Dropdown(
    options=["viridis", "plasma", "inferno", "magma", "cividis"],
    value="plasma",
    description="Colormaps:",
    disabled=False,
)
show_legend = widgets.ToggleButton(
    value=False,
    description="Show legend",
    disabled=False,
    button_style="",  # 'success', 'info', 'warning', 'danger' or ''
    tooltip="Is show legend",
)

def plot(b, cmap, show_legend):
    (
        ggplot("diamonds", aes(x="price"))
        + geom_histogram(bins=b, fill="cut", cmap=cmap)
        + facet_wrap("color", legend=show_legend)
    )

interact(plot, b=b, cmap=cmap, show_legend=show_legend)

<function __main__.plot(b, cmap, show_legend)>