robust-mixed-dist 0.1.0__py3-none-any.whl

robust_mixed_dist/binary.py

@@ -0,0 +1,110 @@
+import polars as pl
+import pandas as pd
+from scipy.spatial import distance
+from scipy.spatial.distance import pdist, squareform
+
+################################################################################
+
+def sokal_dist_matrix(X):
+    """
+    Calculates the Sokal distance matrix for a data matrix `X` using SciPy.
+
+    Parameters (inputs)
+    ----------
+    X: a pandas/polars DataFrame or a NumPy array. It represents a data matrix.
+
+    Returns (outputs)
+    -------
+    M: the Sokal distance matrix between the rows of X.
+    """
+        
+    if isinstance(X, pl.DataFrame):
+        X = X.to_numpy()
+    if isinstance(X, pd.DataFrame):
+        X = X.to_numpy()    
+
+    # Compute the pairwise distances using pdist and convert to a square form.
+    M = squareform(pdist(X, metric='sokalmichener'))
+    
+    return M
+
+################################################################################
+
+def sokal_dist(xi, xr) :
+    """
+    Calculates the Sokal distance between a pair of vectors.
+
+    Parameters (inputs)
+    ----------
+    xi, xr: a pair of quantitative vectors. They represent a couple of statistical observations.
+
+    Returns (outputs)
+    -------
+    The Sokal distance between the observations `xi` and `xr`.
+    """
+
+    if isinstance(xi, (pl.DataFrame, pd.DataFrame)) :
+        xi = xi.to_numpy().flatten()
+    elif isinstance(xi, (pd.Series, pl.Series)) :
+        xi = xi.to_numpy() 
+    if isinstance(xr, (pl.DataFrame, pd.DataFrame)) :
+        xr = xr.to_numpy().flatten()
+    elif isinstance(xr, (pd.Series, pl.Series)) :
+        xr = xr.to_numpy() 
+
+    return distance.sokalmichener(xi, xr)
+
+################################################################################
+
+def jaccard_dist_matrix(X):
+    """
+    Calculates the Jaccard distance matrix for a data matrix `X` using SciPy.
+
+    Parameters (inputs)
+    ----------
+    X: a pandas/polars DataFrame or a NumPy array. It represents a data matrix.
+
+    Returns (outputs)
+    -------
+    M: the Jaccard distance matrix between the rows of X.
+    """
+
+    if isinstance(X, pl.DataFrame):
+        X = X.to_numpy()
+    if isinstance(X, pd.DataFrame):
+        X = X.to_numpy()    
+
+    # Compute the pairwise distances using pdist and convert to a square form.
+    M = squareform(pdist(X, metric='jaccard'))
+    
+    return M
+
+################################################################################
+
+def jaccard_dist(xi, xr) :
+    """
+    Calculates the Jaccard distance between a pair of vectors.
+
+    Parameters (inputs)
+    ----------
+    xi, xr: a pair of quantitative vectors. They represent a couple of statistical observations.
+
+    Returns (outputs)
+    -------
+    The Jaccard distance between the observations `xi` and `xr`.
+    """
+
+    if isinstance(xi, (pl.DataFrame, pd.DataFrame)) :
+        xi = xi.to_numpy().flatten()
+    elif isinstance(xi, (pd.Series, pl.Series)) :
+        xi = xi.to_numpy() 
+    if isinstance(xr, (pl.DataFrame, pd.DataFrame)) :
+        xr = xr.to_numpy().flatten()
+    elif isinstance(xr, (pd.Series, pl.Series)) :
+        xr = xr.to_numpy() 
+        
+    return distance.jaccard(xi, xr)
+
+################################################################################
+
+