line length issue

Author: eyrei123

polars_v_pandas/dataframe_and_lazyframe_time_tests.py

@@ -20,7 +20,9 @@ def create_polars_lazyframe(test_data):
 
 
 def analyze_pandas_dataframe(pandas_df):
-    pandas_df.groupby(["region", "product", "sales_person"])["sales_income"].sum()
+    pandas_df.groupby(
+        ["region", "product", "sales_person"]
+    )["sales_income"].sum()
 
 
 def analyze_polars_dataframe(polars_df):
@@ -38,13 +40,19 @@ def analyze_polars_lazyframe(polars_lf):
 test_data = data_generation(int(sys.argv[1]))
 
 print(f"Pandas dataframe creation time for {int(sys.argv[1])} rows:")
-print(Timer(functools.partial(create_pandas_dataframe, test_data)).timeit(100))
+print(
+    Timer(functools.partial(create_pandas_dataframe, test_data)).timeit(100)
+)
 print()
 print(f"Polars dataframe creation time for {int(sys.argv[1])} rows:")
-print(Timer(functools.partial(create_polars_dataframe, test_data)).timeit(100))
+print(
+    Timer(functools.partial(create_polars_dataframe, test_data)).timeit(100)
+)
 print()
 print(f"Polars lazyframe creation time for {int(sys.argv[1])} rows:")
-print(Timer(functools.partial(create_polars_lazyframe, test_data)).timeit(100))
+print(
+    Timer(functools.partial(create_polars_lazyframe, test_data)).timeit(100)
+)
 
 print()
 
@@ -53,12 +61,18 @@ def analyze_polars_lazyframe(polars_lf):
 polars_lf = create_polars_lazyframe(test_data)
 
 print(f"Pandas dataframe analysis time for {int(sys.argv[1])} rows:")
-print(Timer(functools.partial(analyze_pandas_dataframe, pandas_df)).timeit(100))
+print(
+    Timer(functools.partial(analyze_pandas_dataframe, pandas_df)).timeit(100)
+)
 
 print()
 print(f"Polars dataframe analysis time for {int(sys.argv[1])} rows:")
-print(Timer(functools.partial(analyze_polars_dataframe, polars_df)).timeit(100))
+print(
+    Timer(functools.partial(analyze_polars_dataframe, polars_df)).timeit(100)
+)
 
 print()
 print(f"Polars lazyframe analysis time for {int(sys.argv[1])} rows:")
-print(Timer(functools.partial(analyze_polars_lazyframe, polars_lf)).timeit(100))
+print(
+    Timer(functools.partial(analyze_polars_lazyframe, polars_lf)).timeit(100)
+)

polars_v_pandas/sample_pandas_and_polars_code.py

@@ -19,10 +19,10 @@
 
 
 (
-    orders_pandas
-    .assign(Total=orders_pandas["Quantity"] * orders_pandas["UnitPrice"])
-    .filter(["InvoiceNo", "Quantity", "UnitPrice", "Total"])
-#   .query("Total > 100")
+    orders_pandas.assign(
+        Total=orders_pandas["Quantity"] * orders_pandas["UnitPrice"]
+    ).filter(["InvoiceNo", "Quantity", "UnitPrice", "Total"])
+    #   .query("Total > 100")
 ).head(3)