hoodini.utils.merge_helpers

Lightweight dataframe merge helpers.

View Source

 1"""Lightweight dataframe merge helpers."""
 2
 3from __future__ import annotations
 4
 5import polars as pl
 6
 7
 8def merge_cluster_result(all_prots: pl.DataFrame, merged: pl.DataFrame) -> pl.DataFrame:
 9    if "fam_cluster" in merged.columns:
10        merged = merged.rename({"fam_cluster": "fam_cluster"})
11    return all_prots.join(merged, on="id", how="left")

def merge_cluster_result( all_prots: polars.dataframe.frame.DataFrame, merged: polars.dataframe.frame.DataFrame) -> polars.dataframe.frame.DataFrame: View Source

 9def merge_cluster_result(all_prots: pl.DataFrame, merged: pl.DataFrame) -> pl.DataFrame:
10    if "fam_cluster" in merged.columns:
11        merged = merged.rename({"fam_cluster": "fam_cluster"})
12    return all_prots.join(merged, on="id", how="left")