spells-mtg 0.9.8__py3-none-any.whl → 0.10.0__py3-none-any.whl

spells/utils.py

@@ -0,0 +1,55 @@
+import polars as pl
+
+def convert_to_expr_list(
+    input: str | pl.Expr | list[str | pl.Expr] | None
+):
+    if input is None:
+        return []
+
+    input_list = [input] if isinstance(input, str | pl.Expr) else input
+    return [pl.col(i) if isinstance(i, str) else i for i in input_list]
+
+
+def wavg(
+    df: pl.DataFrame, 
+    cols: str | pl.Expr | list[str | pl.Expr],
+    weights: str | pl.Expr | list[str | pl.Expr],
+    group_by: str | pl.Expr | list[str | pl.Expr] | None = None,
+    new_names: str | list[str] | None = None,
+) -> pl.DataFrame:
+    col_list = convert_to_expr_list(cols)
+    weight_list = convert_to_expr_list(weights)
+    gbs = convert_to_expr_list(group_by)
+
+    name_list: list[str]
+    if isinstance(new_names, str):
+        name_list = [new_names]
+    elif new_names is None:
+        name_list = [c.meta.output_name() for c in col_list]
+    else:
+        name_list = list(new_names)
+
+    assert len(name_list) == len(col_list), f"{len(name_list)} names provided for {len(col_list)} columns"
+    assert len(name_list) == len(set(name_list)), "Output names must be unique"
+    assert len(weight_list) == len(col_list) or len(weight_list) == 1, f"{len(weight_list)} weights provided for {len(col_list)} columns" 
+
+    enum_wl = weight_list * int(len(col_list) / len(weight_list))
+    wl_names = [w.meta.output_name() for w in weight_list]
+    assert len(wl_names) == len(set(wl_names)), "Weights must have unique names. Send one weight column or n uniquely named ones"
+    
+    to_group = df.select(gbs + weight_list + [ 
+        (c * enum_wl[i]).alias(name_list[i]) for i, c in enumerate(col_list)
+    ]) 
+
+    grouped = to_group if not gbs else to_group.group_by(gbs)
+
+    ret_df = grouped.sum().select(
+        gbs + 
+        wl_names + 
+        [(pl.col(name) / pl.col(enum_wl[i].meta.output_name())) for i, name in enumerate(name_list)]
+    )
+   
+    if gbs:
+        ret_df = ret_df.sort(by=gbs)
+
+    return ret_df

{spells_mtg-0.9.8.dist-info → spells_mtg-0.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spells-mtg
-Version: 0.9.8
+Version: 0.10.0
 Summary: analaysis of 17Lands.com public datasets
 Author-Email: Joel Barnes <oelarnes@gmail.com>
 License: MIT

{spells_mtg-0.9.8.dist-info → spells_mtg-0.10.0.dist-info}/RECORD

@@ -11,8 +11,9 @@ spells/filter.py,sha256=J-YTOOAzOQpvIX29tviYL04RVoOUlfsbjBXoQBDCEdQ,3380
 spells/log.py,sha256=3avmg65hru8K9npKLvPp1wWWxq-hoEYDUCbxqhPkKUw,2175
 spells/manifest.py,sha256=dOUmj2uZZ17vCWpFwv7B5F6wOIWnoQdZkEB9SDKdx9M,8310
 spells/schema.py,sha256=DbMvV8PIThJTp0Xzp_XIorlW6JhE1ud1kWRGf5SQ4_c,6406
-spells_mtg-0.9.8.dist-info/METADATA,sha256=TApjATiI8kkKGHCEEEFNhKdmtxUENMD-Qr1VLtDqMic,47007
-spells_mtg-0.9.8.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-spells_mtg-0.9.8.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
-spells_mtg-0.9.8.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
-spells_mtg-0.9.8.dist-info/RECORD,,
+spells/utils.py,sha256=IO3brrXVvZla0LRTEB5v6NgGqZb_rYA46XtKBURGMNk,1944
+spells_mtg-0.10.0.dist-info/METADATA,sha256=Uj4GOhiHcGnWlI31o52ZGZ8PnnZGJggw81N558btGPE,47008
+spells_mtg-0.10.0.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+spells_mtg-0.10.0.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
+spells_mtg-0.10.0.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
+spells_mtg-0.10.0.dist-info/RECORD,,