From 16add70177466e307ce788e74b6847507f47c84f Mon Sep 17 00:00:00 2001 From: ritchie Date: Sat, 15 Jun 2024 16:56:36 +0200 Subject: [PATCH] count supplier --- queries/polars/q21.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/queries/polars/q21.py b/queries/polars/q21.py index 375db28..ba4ecbc 100644 --- a/queries/polars/q21.py +++ b/queries/polars/q21.py @@ -15,8 +15,8 @@ def q() -> None: q1 = ( lineitem.group_by("l_orderkey") - .agg(pl.col("l_suppkey").n_unique().alias("nunique_col")) - .filter(pl.col("nunique_col") > 1) + .agg(pl.col("l_suppkey").len().alias("n_supp_by_order")) + .filter(pl.col("n_supp_by_order") > 1) .join( lineitem.filter(pl.col("l_receiptdate") > pl.col("l_commitdate")), on="l_orderkey", @@ -25,12 +25,12 @@ def q() -> None: q_final = ( q1.group_by("l_orderkey") - .agg(pl.col("l_suppkey").n_unique().alias("nunique_col")) + .agg(pl.col("l_suppkey").len().alias("n_supp_by_order")) .join(q1, on="l_orderkey") .join(supplier, left_on="l_suppkey", right_on="s_suppkey") .join(nation, left_on="s_nationkey", right_on="n_nationkey") .join(orders, left_on="l_orderkey", right_on="o_orderkey") - .filter(pl.col("nunique_col") == 1) + .filter(pl.col("n_supp_by_order") == 1) .filter(pl.col("n_name") == var1) .filter(pl.col("o_orderstatus") == "F") .group_by("s_name")