average_lifetime.py

import pandas as pd

# define topological features to track
homology_dimensions = [0, 1, 2]

# calculate persistence diagram
persistence_diagram = ...

# convert NumPy array of triples to DataFrame
persistence_table = pd.DataFrame(
    persistence_diagram, columns=["birth", "death", "homology_dim"]
)

# calculate the lifetime per point
persistence_table["lifetime"] = persistence_table["death"] - persistence_table["birth"]

# calculate average lifetimes per homology dimension
average_lifetimes = [
    persistence_table.query(f"homology == {homology_dim}")["lifetime"].mean()
    for homology_dim in homology_dimensions
]