sequenzo 0.1.21__cp310-cp310-macosx_11_0_arm64.whl

sequenzo/prefix_tree/system_level_indicators.py

@@ -0,0 +1,465 @@
+"""
+@Author  : Yuqi Liang 梁彧祺
+@File    : system_level_indicators.py
+@Time    : 02/05/2025 11:06
+@Desc    : 
+    This module includes tools for building prefix trees, computing prefix counts, branching factors, and Jensen-Shannon divergence,
+    as well as generating composite scores to summarize system-level sequence diversity and complexity over time.
+    Visualization functions are also provided to plot these indicators and their distributions, 
+    supporting comprehensive analysis of sequence system dynamics.
+"""
+from collections import defaultdict, Counter
+import numpy as np
+from scipy.stats import zscore
+from numpy import array
+from scipy.spatial.distance import jensenshannon
+
+from sequenzo.visualization.utils import save_and_show_results
+import matplotlib.pyplot as plt
+import seaborn as sns
+from typing import List, Optional, Dict
+
+
+class PrefixTree:
+    def __init__(self):
+        self.root = {}
+        self.counts = defaultdict(int)  # prefix -> count
+        self.total_sequences = 0
+
+    def insert(self, sequence):
+        prefix = []
+        node = self.root
+        for state in sequence:
+            prefix.append(state)
+            key = tuple(prefix)
+            self.counts[key] += 1
+            if state not in node:
+                node[state] = {}
+            node = node[state]
+
+    def get_prefixes_at_depth(self, depth):
+        return [k for k in self.counts if len(k) == depth]
+
+    def get_children(self, prefix):
+        """
+        Given a prefix (as a list or tuple), return its immediate children in the tree.
+
+        Returns:
+            dict: mapping from child state -> subtree dict
+        """
+        node = self.root
+        for state in prefix:
+            node = node.get(state, {})
+        return node
+
+    def get_children_count(self, prefix):
+        node = self.root
+        for state in prefix:
+            node = node.get(state, {})
+        return len(node)
+
+    def describe(self):
+        depths = [len(k) for k in self.counts.keys()]
+        max_depth = max(depths) if depths else 0
+        total_prefixes = len(self.counts)
+        print("\n[PrefixTree Overview]")
+        print(f"[>] Total sequences inserted: {self.total_sequences}")
+        print(f"[>] Max depth (time points): {max_depth}")
+        print(f"[>] Total distinct prefixes: {total_prefixes}")
+
+        for t in range(1, max_depth + 1):
+            level_prefixes = self.get_prefixes_at_depth(t)
+            print(f"    Level {t}: {len(level_prefixes)} unique prefixes")
+
+    def __repr__(self):
+        """
+        Returns a brief textual summary of the prefix tree object.
+
+        Note:
+            This method is intended to provide a lightweight, one-line overview
+            (e.g., max depth and total prefix count). For a full structural report
+            including per-level statistics, use the `.describe()` method instead.
+        """
+        depths = [len(k) for k in self.counts.keys()]
+        return f"PrefixTree(max_depth={max(depths) if depths else 0}, total_prefixes={len(self.counts)})"
+
+
+def compute_prefix_count(tree, max_depth):
+    return [len(tree.get_prefixes_at_depth(t)) for t in range(1, max_depth + 1)]
+
+
+def compute_branching_factor(tree, max_depth):
+    result = []
+    for t in range(2, max_depth + 1):
+        prefixes = tree.get_prefixes_at_depth(t - 1)
+        if not prefixes:
+            result.append(0)
+            continue
+        child_counts = [tree.get_children_count(p) for p in prefixes]
+        result.append(np.mean(child_counts))
+    return [0] + result  # pad to align with prefix count
+
+
+def compute_js_divergence(sequences, state_set):
+    T = len(sequences[0])
+    distros = []
+    for t in range(T):
+        counter = Counter(seq[t] for seq in sequences)
+        dist = np.array([counter[s] for s in state_set], dtype=float)
+        dist = dist / dist.sum()
+        distros.append(dist)
+
+    js_scores = [0.0]
+    for t in range(1, T):
+        js = jensenshannon(distros[t], distros[t - 1])
+        js_scores.append(js)
+    return js_scores
+
+
+def build_prefix_tree(sequences):
+    tree = PrefixTree()
+    tree.total_sequences = len(sequences)
+    for seq in sequences:
+        for t in range(1, len(seq) + 1):
+            tree.insert(seq[:t])
+    return tree
+
+
+def plot_system_indicators(
+    prefix_counts: List[float],
+    branching_factors: List[float],
+    js_divergence: Optional[List[float]] = None,
+    x_values: Optional[List] = None,
+    x_label: str = "Time (t)",
+    legend_loc: str = 'lower right',
+    legend_fontsize: int = 10,
+    save_as: Optional[str] = None,
+    figsize: Optional[tuple] = None,
+    dpi: int = 300,
+    custom_colors: Optional[Dict[str, str]] = None,
+    show: bool = True,
+    plot_distributions: bool = False,
+    style: Optional[str] = None
+) -> None:
+    """
+    Plot a single group's system-level indicators using the same visual style as
+    `plot_system_indicators_multiple_comparison`, but for one subplot.
+
+    Design:
+    - Left y-axis: raw Prefix Count
+    - Right y-axis: z-score of Branching Factor and (optionally) JS Divergence
+    - Consistent colors/markers and legend handling with the multi-comparison API
+
+    Parameters:
+    - prefix_counts: List[float]
+        Raw prefix counts per time step
+    - branching_factors: List[float]
+        Branching factor per time step
+    - js_divergence: Optional[List[float]]
+        JS divergence per time step; if None, only branching factor is shown on right axis
+    - x_values: Optional[List]
+        Custom x-axis ticks (e.g., years). If None, uses 1..T. Length must equal data length
+    - x_label: str
+        Label for x-axis. Default: "Time (t)"
+    - legend_loc: str
+        Legend location, e.g., 'upper left', 'upper right', 'lower right', 'best', etc. Default: 'lower right'
+    - legend_fontsize: int
+        Font size for legend text. Default: 10
+    - save_as: Optional[str]
+        If provided, save the figure to this path (png). DPI controlled by `dpi`
+    - figsize: Optional[tuple]
+        Figure size (width, height). Default: (12, 6)
+    - dpi: int
+        Figure DPI when saving. Default: 300
+    - custom_colors: Optional[Dict[str, str]]
+        Optional color overrides. Keys: "Prefix Count", "Branching Factor", "JS Divergence"
+    - show: bool
+        Whether to display the figure
+    - plot_distributions: bool
+        If True, additionally show raw distributions (histograms) of indicators
+    - style: Optional[str]
+        Matplotlib/seaborn style to apply. Common options: 'whitegrid', 'darkgrid', 
+        'white', 'dark', 'ticks'. If None, uses default style. Default: None
+
+    Example:
+    >>> plot_system_indicators(
+    ...     prefix_counts=india_prefix_counts,
+    ...     branching_factors=india_branching_factors,
+    ...     js_divergence=india_js_scores,
+    ...     x_values=[2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019],
+    ...     x_label="Year",
+    ...     legend_loc="lower right",
+    ...     figsize=(12, 6),
+    ...     dpi=300,
+    ... )
+    """
+    T = len(prefix_counts)
+    # Set x values to align with multi-group API
+    if x_values is None:
+        x_values = list(range(1, T + 1))
+    if len(x_values) != T:
+        raise ValueError("Length of x_values must match data length")
+
+    # Normalize others
+    bf_z = zscore(array(branching_factors))
+    js_z = zscore(array(js_divergence)) if js_divergence else None
+
+    color_defaults = {
+        "Prefix Count": "#6BB6FF",    # Soft sky blue (like Monet's water lilies)
+        "Branching Factor": "#FFB347", # Warm peach/coral (like sunset reflections)
+        "JS Divergence": "#F4A6CD",   # Soft rose pink (divergence = different paths)
+    }
+    colors = {**color_defaults, **(custom_colors or {})}
+
+    # --- Main line plot with dual axes ---
+    if figsize is None:
+        figsize = (12, 6)
+    
+    # Apply style if specified
+    if style is not None:
+        # Check if it's a seaborn style
+        seaborn_styles = ['whitegrid', 'darkgrid', 'white', 'dark', 'ticks']
+        if style in seaborn_styles:
+            sns.set_style(style)
+        else:
+            plt.style.use(style)
+    
+    fig, ax1 = plt.subplots(figsize=figsize)
+    ax1.set_xlabel(x_label)
+    ax1.set_ylabel("Prefix Count", color=colors["Prefix Count"])
+    ax1.plot(x_values, prefix_counts, marker='o', color=colors["Prefix Count"], label="Prefix Count")
+    ax1.tick_params(axis='y', labelcolor=colors["Prefix Count"])
+
+    ax2 = ax1.twinx()
+    ax2.set_ylabel("Z-score (Other Indicators)")
+    ax2.plot(x_values, bf_z, marker='s', label='Branching Factor (z)', color=colors["Branching Factor"])
+    if js_z is not None:
+        ax2.plot(x_values, js_z, marker='^', label='JS Divergence (z)', color=colors["JS Divergence"])
+
+    lines1, labels1 = ax1.get_legend_handles_labels()
+    lines2, labels2 = ax2.get_legend_handles_labels()
+    ax2.legend(lines1 + lines2, labels1 + labels2, loc=legend_loc, fontsize=legend_fontsize)
+
+    ax1.set_title("System-Level Trajectory Indicators: Raw vs. Normalized")
+    fig.tight_layout()
+
+    save_and_show_results(save_as=save_as, dpi=dpi, show=show)
+
+    # --- Distribution plots if requested ---
+    if plot_distributions:
+        raw_data = {
+            "Prefix Count": prefix_counts,
+            "Branching Factor": branching_factors,
+        }
+        if js_divergence:
+            raw_data["JS Divergence"] = js_divergence
+
+        n = len(raw_data)
+        fig, axes = plt.subplots(1, n, figsize=(4 * n, 4))
+        if n == 1:
+            axes = [axes]
+
+        for ax, (label, values) in zip(axes, raw_data.items()):
+            sns.histplot(values, kde=True, ax=ax, color=colors.get(label, None))
+            ax.set_title(f"{label} Distribution")
+            ax.set_xlabel("Value")
+            ax.set_ylabel("Density")
+
+        fig.tight_layout()
+        suffix = "_distributions" if save_as else None
+        dist_path = save_as.replace(".png", f"{suffix}.png") if save_as else None
+        save_and_show_results(save_as=dist_path, dpi=dpi, show=show)
+
+
+def plot_system_indicators_multiple_comparison(
+    groups_data: Dict[str, Dict[str, List[float]]],
+    group_names: Optional[List[str]] = None,
+    subplot_titles: Optional[List[str]] = None,
+    x_values: Optional[List] = None,
+    x_label: str = "Time (t)",
+    legend_loc: str = 'lower right',
+    legend_fontsize: int = 10,
+    save_as: Optional[str] = None,
+    figsize: Optional[tuple] = None,
+    dpi: int = 300,
+    custom_colors: Optional[Dict[str, str]] = None,
+    show: bool = True,
+    style: Optional[str] = None
+) -> None:
+    """
+    Plot system-level indicators comparison across multiple groups using dual y-axis design.
+    
+    Parameters:
+    -----------
+    groups_data : Dict[str, Dict[str, List[float]]]
+        Dictionary with group names as keys and data dictionaries as values.
+        Each data dict should contain 'prefix_counts', 'branching_factors', and 'js_divergence'.
+        Example: {
+            "Group1": {
+                "prefix_counts": [10, 15, 20, ...],
+                "branching_factors": [1.2, 1.5, 1.8, ...], 
+                "js_divergence": [0.1, 0.2, 0.15, ...]
+            },
+            "Group2": {...}
+        }
+    group_names : Optional[List[str]]
+        Custom names for groups. If None, uses keys from groups_data.
+        Used for default subplot titles if subplot_titles is not provided.
+    subplot_titles : Optional[List[str]]
+        Custom titles for each subplot. If None, uses default format:
+        "{group_name} - System-Level Trajectory Indicators: Raw vs. Normalized"
+    x_values : Optional[List]
+        Custom x-axis values. If None, uses 1, 2, 3, ...
+    x_label : str
+        Label for x-axis. Default: "Time (t)"
+    legend_loc : str
+        Legend location. Options: 'upper left', 'upper right', 'lower left', 
+        'lower right', 'center', 'best', etc. Default: 'lower right'
+    legend_fontsize : int
+        Font size for legend text. Default: 10
+    save_as : Optional[str]
+        File path to save the plot (without extension)
+    figsize : Optional[tuple]
+        Figure size (width, height). If None, auto-calculated based on number of groups
+    dpi : int
+        DPI for saving. Default: 300
+    custom_colors : Optional[Dict[str, str]]
+        Custom colors for indicators. Default uses standard colors.
+    show : bool
+        Whether to show the plot. Default: True
+    style : Optional[str]
+        Style to apply. Seaborn styles ('whitegrid', 'darkgrid', 'white', 'dark', 'ticks') 
+        or matplotlib styles. If None, uses default style. Default: None
+        
+    Example:
+    --------
+    >>> data = {
+    ...     "India": {
+    ...         "prefix_counts": india_prefix_counts,
+    ...         "branching_factors": india_branching_factors,
+    ...         "js_divergence": india_js_divergence
+    ...     },
+    ...     "US": {
+    ...         "prefix_counts": us_prefix_counts,
+    ...         "branching_factors": us_branching_factors,
+    ...         "js_divergence": us_js_divergence
+    ...     }
+    ... }
+    >>> plot_system_indicators_multiple_comparison(
+    ...     groups_data=data,
+    ...     x_label="Years",
+    ...     legend_loc='upper right',
+    ...     save_as="multi_country_comparison"
+    ... )
+    
+    >>> # With custom subplot titles
+    >>> plot_system_indicators_multiple_comparison(
+    ...     groups_data=data,
+    ...     subplot_titles=["印度发展轨迹", "美国发展轨迹"],
+    ...     x_label="年份",
+    ...     save_as="custom_titles_comparison"
+    ... )
+    """
+    
+    # Validate input
+    if not groups_data:
+        raise ValueError("groups_data cannot be empty")
+    
+    # Get group names
+    if group_names is None:
+        group_names = list(groups_data.keys())
+    
+    if len(group_names) != len(groups_data):
+        raise ValueError("Length of group_names must match number of groups in groups_data")
+    
+    # Validate subplot_titles
+    if subplot_titles is not None and len(subplot_titles) != len(groups_data):
+        raise ValueError("Length of subplot_titles must match number of groups in groups_data")
+    
+    # Get first group to determine data length
+    first_group_data = list(groups_data.values())[0]
+    T = len(first_group_data['prefix_counts'])
+    
+    # Set x values
+    if x_values is None:
+        x_values = list(range(1, T + 1))
+    
+    if len(x_values) != T:
+        raise ValueError("Length of x_values must match data length")
+    
+    # Color settings - Monet-inspired watercolor palette for divergence analysis
+    color_defaults = {
+        "Prefix Count": "#6BB6FF",    # Soft sky blue (like Monet's water lilies)
+        "Branching Factor": "#FFB347", # Warm peach/coral (like sunset reflections)
+        "JS Divergence": "#F4A6CD",   # Soft rose pink (divergence = different paths)
+    }
+    colors = {**color_defaults, **(custom_colors or {})}
+    
+    # Calculate figure size
+    n_groups = len(groups_data)
+    if figsize is None:
+        figsize = (12, 4 * n_groups + 2)  # Dynamic height based on number of groups
+    
+    # Apply style if specified
+    if style is not None:
+        # Check if it's a seaborn style
+        seaborn_styles = ['whitegrid', 'darkgrid', 'white', 'dark', 'ticks']
+        if style in seaborn_styles:
+            sns.set_style(style)
+        else:
+            plt.style.use(style)
+    
+    # Create subplots
+    fig, axes = plt.subplots(n_groups, 1, figsize=figsize)
+    
+    # Handle single group case
+    if n_groups == 1:
+        axes = [axes]
+    
+    # Plot each group
+    for i, (group_key, group_name) in enumerate(zip(groups_data.keys(), group_names)):
+        data = groups_data[group_key]
+        ax = axes[i]
+        
+        # Validate data completeness
+        required_keys = ['prefix_counts', 'branching_factors', 'js_divergence']
+        for key in required_keys:
+            if key not in data:
+                raise ValueError(f"Missing '{key}' in data for group '{group_key}'")
+        
+        # Normalize data (z-score)
+        bf_z = zscore(array(data['branching_factors']))
+        js_z = zscore(array(data['js_divergence']))
+        
+        # Left y-axis: raw prefix counts
+        ax.set_ylabel("Prefix Count", color=colors["Prefix Count"])
+        ax.plot(x_values, data['prefix_counts'], marker='o', 
+                color=colors["Prefix Count"], label="Prefix Count")
+        ax.tick_params(axis='y', labelcolor=colors["Prefix Count"])
+        
+        # Right y-axis: normalized indicators
+        ax_twin = ax.twinx()
+        ax_twin.set_ylabel("Z-score (Other Indicators)")
+        ax_twin.plot(x_values, bf_z, marker='s', 
+                     label='Branching Factor (z)', color=colors["Branching Factor"])
+        ax_twin.plot(x_values, js_z, marker='^', 
+                     label='JS Divergence (z)', color=colors["JS Divergence"])
+        
+        # Legend
+        lines1, labels1 = ax.get_legend_handles_labels()
+        lines2, labels2 = ax_twin.get_legend_handles_labels()
+        ax_twin.legend(lines1 + lines2, labels1 + labels2, loc=legend_loc, fontsize=legend_fontsize)
+        
+        # Title and labels
+        if subplot_titles is not None:
+            title = subplot_titles[i]
+        else:
+            title = f"{group_name} - System-Level Trajectory Indicators: Raw vs. Normalized"
+        ax.set_title(title)
+        
+        # Only set x-label for the bottom subplot
+        if i == n_groups - 1:
+            ax.set_xlabel(x_label)
+    
+    plt.tight_layout()
+    save_and_show_results(save_as=save_as, dpi=dpi, show=show)

sequenzo/prefix_tree/utils.py

@@ -0,0 +1,54 @@
+"""
+@Author  : Yuqi Liang 梁彧祺
+@File    : utils.py
+@Time    : 02/05/2025 12:26
+@Desc    : 
+"""
+import pandas as pd
+from typing import List, Tuple
+
+
+def extract_sequences(df: pd.DataFrame, time_cols: List[str]) -> List[List[str]]:
+    """
+    Efficiently extracts sequences from specified time columns.
+
+    Parameters:
+        df (pd.DataFrame): Input DataFrame.
+        time_cols (List[str]): Columns representing the sequence over time.
+
+    Returns:
+        List[List[str]]: List of sequences (each sequence is a list of states).
+    """
+    return df[time_cols].values.tolist()
+
+
+def get_state_space(sequences: List[List[str]]) -> List[str]:
+    """
+    Efficiently extracts unique states from a list of sequences.
+
+    Parameters:
+        sequences (List[List[str]]): Sequence data.
+
+    Returns:
+        List[str]: Sorted list of unique states.
+    """
+    seen = set()
+    for seq in sequences:
+        seen.update(seq)
+    return sorted(seen)
+
+
+def convert_to_prefix_tree_data(df: pd.DataFrame, time_cols: List[str]) -> Tuple[List[List[str]], List[str]]:
+    """
+    Wrapper to extract sequences and their state space from a DataFrame.
+
+    Parameters:
+        df (pd.DataFrame): Input DataFrame.
+        time_cols (List[str]): Sequence columns (e.g., ['C1', ..., 'C10'])
+
+    Returns:
+        Tuple[List[List[str]], List[str]]: sequences, unique states
+    """
+    sequences = df[time_cols].values.tolist()
+    states = get_state_space(sequences)
+    return sequences, states

sequenzo/sequence_characteristics/__init__.py

@@ -0,0 +1,40 @@
+"""
+@Author  : Yuqi Liang 梁彧祺
+@File    : __init__.py
+@Time    : 22/09/2025 22:46
+@Desc    :
+"""
+from .simple_characteristics import (get_subsequences_in_single_sequence, 
+                                     get_subsequences_all_sequences, 
+                                     get_number_of_transitions)
+
+from .state_frequencies_and_entropy_per_sequence import get_state_freq_and_entropy_per_seq
+
+from .within_sequence_entropy import get_within_sequence_entropy
+
+from .overall_cross_sectional_entropy import get_cross_sectional_entropy
+
+from .variance_of_spell_durations import get_spell_duration_variance
+
+from .turbulence import get_turbulence
+
+from .complexity_index import get_complexity_index
+
+from .plot_characteristics import plot_longitudinal_characteristics, plot_cross_sectional_characteristics
+
+__all__ = [
+    "get_subsequences_in_single_sequence",
+    "get_subsequences_all_sequences",
+    "get_number_of_transitions",
+
+    "get_complexity_index",
+
+    "get_state_freq_and_entropy_per_seq",
+    "get_within_sequence_entropy",
+    "get_cross_sectional_entropy",
+    "get_spell_duration_variance",
+    "get_turbulence",
+
+    "plot_longitudinal_characteristics",
+    "plot_cross_sectional_characteristics"
+]

sequenzo/sequence_characteristics/complexity_index.py

@@ -0,0 +1,49 @@
+"""
+@Author  : 李欣怡
+@File    : complexity_index.py
+@Time    : 2025/9/23 23:45
+@Desc    : Complexity index
+
+        The corresponding function name in TraMineR is seqici.R,
+        with the source code available at: https://github.com/cran/TraMineR/blob/master/R/seqici.R
+"""
+import os
+from contextlib import redirect_stdout
+import pandas as pd
+import numpy as np
+
+from sequenzo.define_sequence_data import SequenceData
+from .simple_characteristics import get_number_of_transitions
+from .within_sequence_entropy import get_within_sequence_entropy
+
+def get_complexity_index(seqdata, silent=True):
+    if not isinstance(seqdata, SequenceData):
+        raise ValueError("[!] data is NOT a sequence object, see SequenceData function to create one.")
+
+    if not silent:
+        print(f"[>] Computing complexity index for {seqdata.seqdata.shape[0]} sequences ...")
+
+    trans = get_number_of_transitions(seqdata=seqdata, norm=True).iloc[:, 1]
+    trans.index = seqdata.seqdata.index
+
+    with open(os.devnull, 'w') as fnull:
+        with redirect_stdout(fnull):
+            ient = get_within_sequence_entropy(seqdata=seqdata, norm=True).iloc[:, 1]
+            ient.index = seqdata.seqdata.index
+
+    complxity = np.sqrt(trans * ient)
+    complxity = pd.DataFrame(complxity, index=seqdata.seqdata.index, columns=['Complexity Index'])
+    complxity = complxity.reset_index().rename(columns={'index': 'ID'})
+
+    return complxity
+
+if __name__ == '__main__':
+    from sequenzo import *
+
+    df = load_dataset("country_co2_emissions")
+    _time = list(df.columns)[1:]
+    states = ['Very Low', 'Low', 'Middle', 'High', 'Very High']
+    sequence_data = SequenceData(df, time=_time, id_col="country", states=states)
+
+    res = get_complexity_index(sequence_data)
+    res