statslibx 0.1.7__py3-none-any.whl → 0.1.8__py3-none-any.whl

statslibx/__init__.py

@@ -1,30 +1,32 @@
 """
 StatsLibx - Librería de Estadística para Python
 Autor: Emmanuel Ascendra
-Versión: 0.1.6
+Versión: 0.1.8
 """
 
-__version__ = "0.1.6"
+__version__ = "0.1.8"
 __author__ = "Emmanuel Ascendra"
 
 # Importar las clases principales
 from .descriptive import DescriptiveStats, DescriptiveSummary
 from .inferential import InferentialStats, TestResult
+from .probability import ProbabilityStats
+from .computacional import ComputacionalStats
 from .utils import UtilsStats
 from .preprocessing import Preprocessing
-from .datasets import load_dataset
+from .datasets import load_dataset, generate_dataset
 
 # Definir qué se expone cuando se hace: from statslib import *
 __all__ = [
     # Clases principales
     'DescriptiveStats',
-    'InferentialStats', 
-    'LinearRegressionResult',
-    'DescriptiveSummary',
-    'TestResult',
+    'InferentialStats',
+    'ProbabilityStats',
+    'ComputacionalStats',
     'UtilsStats',
     'Preprocessing',
-    'load_dataset'
+    'load_dataset',
+    'generate_dataset'
 ]
 
 # Mensaje de bienvenida (opcional)
@@ -37,5 +39,7 @@ def welcome():
     print(f"  - DescriptiveStats: Estadística descriptiva")
     print(f"  - InferentialStats: Estadística inferencial")
     print(f"  - UtilsStats: Utilidades Extras")
+    print(f"\nMódulos disponibles:")
+    print(f"  - Datasets: Carga de Datasets")
     print(f"  - Preprocessing: Preprocesamiento de datos")
     print(f"\nPara más información: help(statslibx)")

statslibx/computacional.py

@@ -0,0 +1,2 @@
+class ComputacionalStats:
+    pass

statslibx/datasets/__init__.py

@@ -1,71 +1,244 @@
-from typing import Optional, Union, Literal, List
-import polars as pl
-import pandas as pd
-import pkgutil
+from typing import Optional, Union, Literal, List, Tuple
 import io
+import pkgutil
+
+import pandas as pd
+import polars as pl
+import numpy as np
+from numpy.typing import NDArray
+
+
+_SUPPORTED_BACKENDS = ("pandas", "polars")
+
+
+def _validate_columns(
+    df: Union[pd.DataFrame, pl.DataFrame],
+    X_columns: List[str],
+    y_column: str
+) -> None:
+    columns = set(df.columns)
+    missing = set(X_columns + [y_column]) - columns
+    if missing:
+        raise ValueError(f"Columnas no encontradas en el dataset: {missing}")
+
+
+def _X_y(
+    df: Union[pd.DataFrame, pl.DataFrame],
+    X_columns: List[str],
+    y_column: str
+) -> Tuple[NDArray, NDArray]:
+    """
+    Extrae X e y como arrays numpy desde pandas o polars.
+    """
+    _validate_columns(df, X_columns, y_column)
+
+    if isinstance(df, pd.DataFrame):
+        X = df[X_columns].to_numpy()
+        y = df[y_column].to_numpy().ravel()
+        return X, y
+
+    elif isinstance(df, pl.DataFrame):
+        X = df.select(X_columns).to_numpy()
+        y = df.select(y_column).to_numpy().ravel()
+        return X, y
+
+    else:
+        raise TypeError(
+            "Backend no soportado. Use pandas.DataFrame o polars.DataFrame."
+        )
+
 
 def load_dataset(
-        name: str,
-        backend: Literal['pandas', 'polars'] = 'pandas'
-    ) -> Union[pd.DataFrame, pl.DataFrame]:
-    """Carga un dataset interno del paquete.
-    Datasets Disponibles:
+    name: str,
+    backend: Literal["pandas", "polars"] = "pandas",
+    return_X_y: Optional[Tuple[List[str], str]] = None
+) -> Union[pd.DataFrame, pl.DataFrame, Tuple[NDArray, NDArray]]:
+    """
+    Carga un dataset interno del paquete.
+
+    Datasets disponibles:
     - iris.csv
     - penguins.csv
     - sp500_companies.csv
     - titanic.csv
     - course_completion.csv
+
+    Parámetros
+    ----------
+    name : str
+        Nombre del archivo CSV.
+    backend : {'pandas', 'polars'}, default='pandas'
+        Backend de DataFrame a utilizar.
+    return_X_y : tuple[list[str], str], optional
+        Si se especifica, devuelve (X, y) como arrays numpy,
+
+    Retorna
+    -------
+    DataFrame o (X, y)
     """
-    data_bytes = pkgutil.get_data("statslibx.datasets", name)
-    if data_bytes is None:
-        raise FileNotFoundError(f"Dataset '{name}' no encontrado.")
-    
-    if backend == "pandas":
-        return pd.read_csv(io.BytesIO(data_bytes))
-    elif backend == "polars":
-        return pl.read_csv(io.BytesIO(data_bytes))
-    else:
+    if backend not in _SUPPORTED_BACKENDS:
         raise ValueError(
-            "Backend no soportado. Use 'pandas' o 'polars'."
+            f"Backend '{backend}' no soportado. "
+            f"Use uno de {_SUPPORTED_BACKENDS}."
         )
-    
-def load_iris(
-        backend: Literal['pandas', 'polars'] = 'pandas'
-    ) -> Union[pd.DataFrame, pl.DataFrame]:
-    """Carga el dataset interno de la libreria: Iris
-    """
-    data_bytes = pkgutil.get_data("statslibx.datasets", "iris.csv")
-    if data_bytes is None:
-        raise FileNotFoundError(f"Dataset \"iris.csv\" no encontrado.")
-    
-    if backend == "pandas":
-        return pd.read_csv(io.BytesIO(data_bytes))
-    elif backend == "polars":
-        raise ValueError(
-            "Backend no soportado aun. Use 'pandas'."
+
+    # ---------- 1️⃣ Intentar cargar desde el paquete ----------
+    data_bytes = pkgutil.get_data("statslibx.datasets", name)
+
+    if data_bytes is not None:
+        df = (
+            pd.read_csv(io.BytesIO(data_bytes))
+            if backend == "pandas"
+            else pl.read_csv(io.BytesIO(data_bytes))
         )
+
+    # ---------- 2️⃣ Si no está en el paquete, buscar en ruta actual ----------
     else:
-        raise ValueError(
-            "Backend no soportado. Use 'pandas' o 'polars'."
-        )
-    
+        try:
+            df = (
+                pd.read_csv(name)
+                if backend == "pandas"
+                else pl.read_csv(name)
+            )
+        except FileNotFoundError:
+            raise FileNotFoundError(
+                f"Dataset '{name}' no encontrado "
+                f"ni en statslibx.datasets ni en la ruta actual."
+            )
+
+    # ---------- 3️⃣ Devolver X, y si se solicita ----------
+    if return_X_y is not None:
+        X_columns, y_column = return_X_y
+        return _X_y(df, X_columns, y_column)
+
+    return df
+
+# =========================
+# Datasets específicos
+# =========================
+
+def load_iris(
+    backend: Literal["pandas", "polars"] = "pandas",
+    return_X_y: Optional[Tuple[List[str], str]] = None
+):
+    return load_dataset(
+        "iris.csv",
+        backend=backend,
+        return_X_y=return_X_y
+    )
+
+
 def load_penguins(
-        backend: Literal['pandas', 'polars'] = 'pandas'
-    ) -> Union[pd.DataFrame, pl.DataFrame]:
-    """Carga un dataset interno de la libreria: Penguins
-    """
-    data_bytes = pkgutil.get_data("statslibx.datasets", "penguins.csv")
-    if data_bytes is None:
-        raise FileNotFoundError(f"Dataset \"penguins.csv\" no encontrado.")
+    backend: Literal["pandas", "polars"] = "pandas",
+    return_X_y: Optional[Tuple[List[str], str]] = None
+):
+    return load_dataset(
+        "penguins.csv",
+        backend=backend,
+        return_X_y=return_X_y
+    )
+
+
+from typing import Optional
+
+def generate_dataset(n_rows, schema, seed=None, save: Optional[bool] = False, filename: Optional[str] = None):
+    if seed is not None:
+        if not isinstance(seed, int):
+            raise TypeError("seed debe ser un entero o None")
+        np.random.seed(seed)
+    else:
+        np.random.seed(42)
+
+    if not isinstance(schema, dict):
+        raise TypeError("schema debe ser un diccionario")
     
-    if backend == "pandas":
-        return pd.read_csv(io.BytesIO(data_bytes))
-    elif backend == "polars":
-        raise ValueError(
-            "Backend no soportado aun. Use 'pandas'."
-        )
+
+
+    data = {}
+
+    for col, config in schema.items():
+        if "dist" not in config:
+            raise ValueError(f"La columna '{col}' no tiene 'dist' definido")
+
+        dist = config["dist"]
+        dtype = config.get("type", "float")
+        nround = config.get("round", 0)
+
+        # ---------- DISTRIBUCIONES ----------
+        if dist == "normal":
+            values = np.random.normal(
+                loc=config.get("mean", 0),
+                scale=config.get("std", 1),
+                size=n_rows
+            )
+
+        elif dist == "uniform":
+            values = np.random.uniform(
+                low=config.get("low", 0),
+                high=config.get("high", 1),
+                size=n_rows
+            )
+
+        elif dist == "exponential":
+            values = np.random.exponential(
+                scale=config.get("scale", 1),
+                size=n_rows
+            )
+
+        elif dist == "lognormal":
+            values = np.random.lognormal(
+                mean=config.get("mean", 0),
+                sigma=config.get("std", 1),
+                size=n_rows
+            )
+
+        elif dist == "poisson":
+            values = np.random.poisson(
+                lam=config.get("lam", 1),
+                size=n_rows
+            )
+
+        elif dist == "binomial":
+            values = np.random.binomial(
+                n=config.get("n", 1),
+                p=config.get("p", 0.5),
+                size=n_rows
+            )
+
+        elif dist == "categorical":
+            if "choices" not in config:
+                raise ValueError(f"'choices' es requerido para categorical ({col})")
+            values = np.random.choice(
+                config["choices"],
+                size=n_rows
+            )
+            data[col] = values
+            continue
+
+        else:
+            raise ValueError(f"Distribución no soportada: {dist}")
+
+        # ---------- CASTEO DE TIPO ----------
+        if dtype == "int":
+            values = np.round(values).astype(int)
+        elif dtype == "float":
+            values = values.astype(float)
+        else:
+            raise ValueError(f"Tipo no soportado: {dtype}")
+        
+        # ---------- REDONDEO ----------
+        if nround > 0:
+            values = np.round(values, nround)
+        else:
+            values = np.round(values, 2)
+
+        data[col] = values
+
+    if save and filename:
+        df = pd.DataFrame(data)
+        df.to_csv(f"{filename}.csv", index=False)
     else:
-        raise ValueError(
-            "Backend no soportado. Use 'pandas' o 'polars'."
-        )
+        df = pd.DataFrame(data)
+        df.to_csv("dataset.csv", index=False)
 
+    return pd.DataFrame(data)

statslibx/descriptive.py

@@ -13,7 +13,6 @@ import plotly.express as px
 
 class DescriptiveStats:
     """
-    Clase para estadística descriptiva univariada y multivariada
     Class for univariate and multivariate descriptive statistics
     """
     
@@ -23,7 +22,7 @@ class DescriptiveStats:
                 thousand: str = None,
                 backend: Literal['pandas', 'polars'] = 'pandas'):
         """
-        # Inicialize DataFrame
+        # Initialize DataFrame
         
         ## **Parameters:**
 
@@ -54,7 +53,8 @@ class DescriptiveStats:
             if data.ndim == 1:
                 data = pd.DataFrame({'var': data})
             else:
-                data = pd.DataFrame(data, columns=[f'var_{i}' for i in range(data.shape[1])])
+                data = pd.DataFrame(data, columns=[f'var_{i}' for i in range(data.shape[1])]) \
+                    if isinstance(data, pd.DataFrame) else pl.DataFrame(data, )
         
         self.data = data
         self.backend = backend
@@ -394,7 +394,7 @@ class DescriptiveStats:
 
 
     
-    def help(self, lang="es-Es"):
+    def help(self, lang="es-ES"):
         """
         Muestra ayuda completa de la clase DescriptiveStats
 
@@ -407,7 +407,8 @@ class DescriptiveStats:
         if lang in ["en-US", "English", "english"]:
             lang = "en-US"
         else:
-            lang = ""
+            lang = "es-ES"
+        help_text = " "
 
         match lang:
             case "es-ES":
@@ -614,10 +615,9 @@ class DescriptiveStats:
 ╚════════════════════════════════════════════════════════════════════════════╝
     """
             case "en-US":
-                # --- Falta por traducir
                 help_text = """
 ╔════════════════════════════════════════════════════════════════════════════╗
-║                    📊 DescriptiveStats CLASS - COMPLETE HELP               ║
+║                    📊 CLASS DescriptiveStats  - COMPLETE HELP              ║
 ╚════════════════════════════════════════════════════════════════════════════╝
 
 📝 DESCRIPTION:
@@ -821,8 +821,7 @@ class DescriptiveStats:
         
         print(help_text)
 
-    
-    
+
 class DescriptiveSummary:
     """Clase para formatear salida de estadística descriptiva"""