Modify the TPC-H queries of duckdb, pandas, polars, and cuDF (#5)

* Modify the TPC-H queries of duckdb, pandas, polars, and cuDF
godaai · Apr 21, 2024 · fbcfd52 · fbcfd52
1 parent 25b842b
commit fbcfd52
Show file tree

Hide file tree

Showing 6 changed files with 1,626 additions and 274 deletions.
diff --git a/tpch/common_utils.py b/tpch/common_utils.py
@@ -38,6 +38,9 @@ def print_result_fn(solution: str, result: pd.DataFrame, query: str):
     if not os.path.exists(result_prefix):
         os.makedirs(result_prefix)
     result_path = f"{result_prefix}/{query}.out"
+    if(solution == "polars"):
+        result.write_csv(result_path)
+        return
     result.to_csv(result_path, index=False)