feat: add vertical upset plot orientation with horizontal cardinality

Implemented a new 'orientation' parameter for UpSet plots that allows switching between horizontal and vertical layouts:
- Horizontal (default): cardinality bars on top (vertical), set sizes on right (horizontal)
- Vertical (new): cardinality bars on left (horizontal), set sizes on top (vertical)

Changes:
- Added 'orientation' parameter to UpSetAltair() function with validation
- Updated create_vertical_bar() to support both orientations by swapping X/Y axes
- Updated create_matrix_view() to swap axes for vertical orientation
- Updated create_horizontal_bar() to support vertical layout with Y-axis for set sizes
- Refactored layout logic to use hconcat for vertical orientation
- Added comprehensive tests for vertical orientation
- Updated test normalization to handle auto-generated param and view names

All existing tests pass, and the feature is fully backward compatible.

Author: claude

altair_upset/components.py

@@ -12,27 +12,53 @@ def create_vertical_bar(
     tooltip,
     vertical_bar_label_size,
     vertical_bar_y_axis_orient,
+    orientation="horizontal",
 ):
-    """Creates the vertical bar chart component."""
-    vertical_bar = base.mark_bar(color=main_color, size=vertical_bar_size).encode(
-        x=alt.X(
-            "intersection_id:N",
-            axis=alt.Axis(grid=False, labels=False, ticks=False, domain=True),
-            sort=x_sort,
-            title=None,
-        ),
-        y=alt.Y(
-            "max(count):Q",
-            axis=alt.Axis(grid=False, tickCount=3, orient=vertical_bar_y_axis_orient),
-            title="Intersection Size",
-        ),
-        color=brush_color,
-        tooltip=tooltip,
-    )
-
-    vertical_bar_text = vertical_bar.mark_text(
-        color=main_color, dy=-10, size=vertical_bar_label_size
-    ).encode(text=alt.Text("count:Q", format=".0f"))
+    """Creates the vertical bar chart component.
+
+    In horizontal orientation: bars go up (Y-axis shows cardinality)
+    In vertical orientation: bars go right (X-axis shows cardinality)
+    """
+    if orientation == "horizontal":
+        vertical_bar = base.mark_bar(color=main_color, size=vertical_bar_size).encode(
+            x=alt.X(
+                "intersection_id:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=True),
+                sort=x_sort,
+                title=None,
+            ),
+            y=alt.Y(
+                "max(count):Q",
+                axis=alt.Axis(grid=False, tickCount=3, orient=vertical_bar_y_axis_orient),
+                title="Intersection Size",
+            ),
+            color=brush_color,
+            tooltip=tooltip,
+        )
+
+        vertical_bar_text = vertical_bar.mark_text(
+            color=main_color, dy=-10, size=vertical_bar_label_size
+        ).encode(text=alt.Text("count:Q", format=".0f"))
+    else:  # vertical orientation - swap X and Y
+        vertical_bar = base.mark_bar(color=main_color, size=vertical_bar_size).encode(
+            y=alt.Y(
+                "intersection_id:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=True),
+                sort=x_sort,
+                title=None,
+            ),
+            x=alt.X(
+                "max(count):Q",
+                axis=alt.Axis(grid=False, tickCount=3, orient=vertical_bar_y_axis_orient),
+                title="Intersection Size",
+            ),
+            color=brush_color,
+            tooltip=tooltip,
+        )
+
+        vertical_bar_text = vertical_bar.mark_text(
+            color=main_color, dx=10, size=vertical_bar_label_size
+        ).encode(text=alt.Text("count:Q", format=".0f"))
 
     return vertical_bar, vertical_bar_text
 
@@ -45,38 +71,75 @@ def create_matrix_view(
     brush_color,
     line_connection_size,
     main_color,
+    orientation="horizontal",
 ):
-    """Creates the matrix view component."""
-    circle_bg = vertical_bar.mark_circle(size=glyph_size, opacity=1).encode(
-        x=alt.X(
-            "intersection_id:N",
-            axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
-            sort=x_sort,
-            title=None,
-        ),
-        y=alt.Y(
-            "set_order:N",
-            axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
-            title=None,
-        ),
-        color=alt.value("#E6E6E6"),
-    )
-
-    rect_bg = (
-        circle_bg.mark_rect()
-        .transform_filter(alt.datum["set_order"] % 2 == 1)
-        .encode(color=alt.value("#F7F7F7"))
-    )
-
-    circle = circle_bg.transform_filter(alt.datum["is_intersect"] == 1).encode(
-        color=brush_color
-    )
-
-    line_connection = (
-        vertical_bar.mark_bar(size=line_connection_size, color=main_color)
-        .transform_filter(alt.datum["is_intersect"] == 1)
-        .encode(y=alt.Y("min(set_order):N"), y2=alt.Y2("max(set_order):N"))
-    )
+    """Creates the matrix view component.
+
+    In horizontal orientation: X = intersection_id, Y = set_order
+    In vertical orientation: X = set_order, Y = intersection_id
+    """
+    if orientation == "horizontal":
+        circle_bg = vertical_bar.mark_circle(size=glyph_size, opacity=1).encode(
+            x=alt.X(
+                "intersection_id:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
+                sort=x_sort,
+                title=None,
+            ),
+            y=alt.Y(
+                "set_order:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
+                title=None,
+            ),
+            color=alt.value("#E6E6E6"),
+        )
+
+        rect_bg = (
+            circle_bg.mark_rect()
+            .transform_filter(alt.datum["set_order"] % 2 == 1)
+            .encode(color=alt.value("#F7F7F7"))
+        )
+
+        circle = circle_bg.transform_filter(alt.datum["is_intersect"] == 1).encode(
+            color=brush_color
+        )
+
+        line_connection = (
+            vertical_bar.mark_bar(size=line_connection_size, color=main_color)
+            .transform_filter(alt.datum["is_intersect"] == 1)
+            .encode(y=alt.Y("min(set_order):N"), y2=alt.Y2("max(set_order):N"))
+        )
+    else:  # vertical orientation - swap X and Y
+        circle_bg = vertical_bar.mark_circle(size=glyph_size, opacity=1).encode(
+            y=alt.Y(
+                "intersection_id:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
+                sort=x_sort,
+                title=None,
+            ),
+            x=alt.X(
+                "set_order:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
+                title=None,
+            ),
+            color=alt.value("#E6E6E6"),
+        )
+
+        rect_bg = (
+            circle_bg.mark_rect()
+            .transform_filter(alt.datum["set_order"] % 2 == 1)
+            .encode(color=alt.value("#F7F7F7"))
+        )
+
+        circle = circle_bg.transform_filter(alt.datum["is_intersect"] == 1).encode(
+            color=brush_color
+        )
+
+        line_connection = (
+            vertical_bar.mark_bar(size=line_connection_size, color=main_color)
+            .transform_filter(alt.datum["is_intersect"] == 1)
+            .encode(x=alt.X("min(set_order):N"), x2=alt.X2("max(set_order):N"))
+        )
 
     return circle_bg, rect_bg, circle, line_connection
 
@@ -90,34 +153,68 @@ def create_horizontal_bar(
     horizontal_bar_label_bg_color,
     horizontal_bar_size,
     horizontal_bar_chart_width,
+    orientation="horizontal",
 ):
-    """Creates the horizontal bar chart component."""
-    horizontal_bar_label_bg = base.mark_circle(size=set_label_bg_size).encode(
-        y=alt.Y(
-            "set_order:N",
-            axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
-            title=None,
-        ),
-        color=alt.Color(
-            "set:N", scale=alt.Scale(domain=sets, range=color_range), title=None
-        ),
-        opacity=alt.value(1),
-    )
-
-    horizontal_bar_label = horizontal_bar_label_bg.mark_text(
-        align=("center" if is_show_horizontal_bar_label_bg else "center")
-    ).encode(
-        text=alt.Text("set_abbre:N"), color=alt.value(horizontal_bar_label_bg_color)
-    )
-
-    horizontal_bar = (
-        horizontal_bar_label_bg.mark_bar(size=horizontal_bar_size)
-        .transform_filter(alt.datum["is_intersect"] == 1)
-        .encode(
+    """Creates the horizontal bar chart component.
+
+    In horizontal orientation: Y = set_order, X = sum(count) (bars go right)
+    In vertical orientation: X = set_order, Y = sum(count) (bars go up)
+    """
+    if orientation == "horizontal":
+        horizontal_bar_label_bg = base.mark_circle(size=set_label_bg_size).encode(
+            y=alt.Y(
+                "set_order:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
+                title=None,
+            ),
+            color=alt.Color(
+                "set:N", scale=alt.Scale(domain=sets, range=color_range), title=None
+            ),
+            opacity=alt.value(1),
+        )
+
+        horizontal_bar_label = horizontal_bar_label_bg.mark_text(
+            align=("center" if is_show_horizontal_bar_label_bg else "center")
+        ).encode(
+            text=alt.Text("set_abbre:N"), color=alt.value(horizontal_bar_label_bg_color)
+        )
+
+        horizontal_bar = (
+            horizontal_bar_label_bg.mark_bar(size=horizontal_bar_size)
+            .transform_filter(alt.datum["is_intersect"] == 1)
+            .encode(
+                x=alt.X(
+                    "sum(count):Q", axis=alt.Axis(grid=False, tickCount=3), title="Set Size"
+                )
+            )
+        )
+    else:  # vertical orientation - swap X and Y
+        horizontal_bar_label_bg = base.mark_circle(size=set_label_bg_size).encode(
             x=alt.X(
-                "sum(count):Q", axis=alt.Axis(grid=False, tickCount=3), title="Set Size"
+                "set_order:N",
+                axis=alt.Axis(grid=False, labels=False, ticks=False, domain=False),
+                title=None,
+            ),
+            color=alt.Color(
+                "set:N", scale=alt.Scale(domain=sets, range=color_range), title=None
+            ),
+            opacity=alt.value(1),
+        )
+
+        horizontal_bar_label = horizontal_bar_label_bg.mark_text(
+            align=("center" if is_show_horizontal_bar_label_bg else "center")
+        ).encode(
+            text=alt.Text("set_abbre:N"), color=alt.value(horizontal_bar_label_bg_color)
+        )
+
+        horizontal_bar = (
+            horizontal_bar_label_bg.mark_bar(size=horizontal_bar_size)
+            .transform_filter(alt.datum["is_intersect"] == 1)
+            .encode(
+                y=alt.Y(
+                    "sum(count):Q", axis=alt.Axis(grid=False, tickCount=3), title="Set Size"
+                )
             )
         )
-    )
 
     return horizontal_bar_label_bg, horizontal_bar_label, horizontal_bar

altair_upset/upset.py

@@ -71,6 +71,7 @@ def UpSetAltair(
     abbre: Optional[List[str]] = None,
     sort_by: str = "frequency",
     sort_order: str = "ascending",
+    orientation: str = "horizontal",
     width: int = 1200,
     height: int = 700,
     height_ratio: float = 0.6,
@@ -119,6 +120,10 @@ def UpSetAltair(
         - "degree": sort by number of sets in intersection
     sort_order : {"ascending", "descending"}, default "ascending"
         Order of sorting for intersections.
+    orientation : {"horizontal", "vertical"}, default "horizontal"
+        Orientation of the plot:
+        - "horizontal": cardinality bars are vertical (top), set sizes are horizontal (right)
+        - "vertical": cardinality bars are horizontal (left), set sizes are vertical (top)
     width : int, default 1200
         Total width of the plot in pixels.
     height : int, default 700
@@ -200,6 +205,8 @@ def UpSetAltair(
         raise ValueError("if provided, abbre must have the same length as sets")
     if vertical_bar_y_axis_orient not in ["left", "right"]:
         raise ValueError("vertical bar y axis orient must be 'left' or 'right'")
+    if orientation not in ["horizontal", "vertical"]:
+        raise ValueError("orientation must be either 'horizontal' or 'vertical'")
 
     # Apply theme if specified
     if theme is not None:
@@ -261,6 +268,7 @@ def UpSetAltair(
         tooltip,
         vertical_bar_label_size,
         vertical_bar_y_axis_orient,
+        orientation,
     )
     vertical_bar_chart = (
         (vertical_bar + vertical_bar_text)
@@ -276,6 +284,7 @@ def UpSetAltair(
         brush_color,
         line_connection_size,
         main_color,
+        orientation,
     )
     matrix_view = (
         (circle + rect_bg + circle_bg + line_connection + circle)
@@ -293,25 +302,43 @@ def UpSetAltair(
             horizontal_bar_label_bg_color,
             horizontal_bar_size,
             horizontal_bar_chart_width,
+            orientation,
         )
     )
     horizontal_bar_axis = (
         (horizontal_bar_label_bg + horizontal_bar_label)
         if is_show_horizontal_bar_label_bg
         else horizontal_bar_label
-    ).properties(width=horizontal_bar_chart_width)
-
-    # Combine components
-    upsetaltair = alt.vconcat(
-        vertical_bar_chart,
-        alt.hconcat(
-            matrix_view,
-            horizontal_bar_axis,
-            horizontal_bar.properties(width=horizontal_bar_chart_width),
-            spacing=0,  # Minimize spacing between components
-        ).resolve_scale(x="shared", y="shared"),  # X shared also
-        spacing=5,
-    ).add_params(legend_selection)
+    ).properties(width=horizontal_bar_chart_width if orientation == "horizontal" else matrix_width)
+
+    # Combine components based on orientation
+    if orientation == "horizontal":
+        # Horizontal layout: cardinality bars on top, set sizes on right
+        upsetaltair = alt.vconcat(
+            vertical_bar_chart,
+            alt.hconcat(
+                matrix_view,
+                horizontal_bar_axis,
+                horizontal_bar.properties(width=horizontal_bar_chart_width),
+                spacing=0,
+            ).resolve_scale(x="shared", y="shared"),
+            spacing=5,
+        ).add_params(legend_selection)
+    else:
+        # Vertical layout: cardinality bars on left, set sizes on top
+        upsetaltair = alt.hconcat(
+            vertical_bar_chart.properties(width=vertical_bar_chart_height, height=matrix_height),
+            alt.vconcat(
+                alt.hconcat(
+                    horizontal_bar_axis,
+                    horizontal_bar.properties(width=matrix_width, height=horizontal_bar_chart_width),
+                    spacing=0,
+                ).resolve_scale(x="shared", y="shared"),
+                matrix_view.properties(height=matrix_height),
+                spacing=5,
+            ),
+            spacing=5,
+        ).add_params(legend_selection)
 
     # Apply configuration
     chart = upsetaltair_top_level_configuration(