Combining waterfall_plot() with Other Geometry Layers

This notebook demonstrates how to enrich a waterfall plot with background and foreground layers. Foreground layers can be added using the regular + operator. To add background layers, use the new background_layers property.

Limitations:

• Layers must provide their own data
• Data coordinates are expected to be numeric

import pandas as pd

from lets_plot import *
from lets_plot.bistro import waterfall_plot

LetsPlot.setup_html()

# Waterfall data
df = pd.DataFrame({
    "Accounts": ["initial", "revenue", "costs", "revenue", "costs", "revenue", "costs", "revenue", "costs", "total"],
    "Values": [200, 200, -150, 250, -170, 150, -50, 280, -25, None],
    "Measure": ["absolute", "relative", "relative", "relative", "relative", "relative", "relative", "relative",
                "relative", "total"],
})

# Background layer and its data
quarter_data = {
    "period_start": [0.5, 4.5],
    "period_end": [4.5, 8.5],
    "ai_introduced": [False, True],
}
quarter_layer = geom_band(
    aes(
        xmin="period_start",
        xmax="period_end",
        paint_a="ai_introduced"
    ),
    data=quarter_data,
    alpha=0.2,
    # We use "paint_a" to color the bands based on a separate category (e.g., quarters),
    # so they have their own color palette independent from the waterfalls
    fill_by="paint_a",
    color_by="paint_a"
)

# Foreground layers and their data
quarter_label_data = {
    "name": ["Q1", "Q2", "Q3", "Q4"],
    "x": [1.5, 3.5, 5.5, 7.5],
    "y": [750] * 4
}
quarter_ai_status_data = {
    "text": ["Before AI\nintroduction", "After AI\nintroduction"],
    "x": [2.5, 6.5],
    "y": [100, 100],
}
text_layers = geom_text(aes(x="x", y="y", label="name"), data=quarter_label_data, size=8) + \
    geom_text(aes(x="x", y="y", label="text"), data=quarter_ai_status_data, size=12)

# Whole plot
(waterfall_plot(df, "Accounts", "Values", measure="Measure",
                background_layers=quarter_layer)  # Background layer
  + text_layers                                   # Foreground layers
  + scale_hue("paint_a", guide="none")            # Color for the background layer (bands)
  + ggsize(750, 450)
  + ggtitle("Waterfall with additional layers"))