79367410b43a54f3fc7674067fa962ed55b1d6bb,lifelines/fitters/npmle.py,,reconstruct_survival_function,#Any#Any#Any#Any#,128

Before Change

def reconstruct_survival_function(probabilities, turnball_intervals, timeline, label="NPMLE"):
    index = [0.0]
    values = [1.0]

    for p, interval in zip(probabilities, turnball_intervals):
        if interval.left != index[-1]:
            index.append(interval.left)

After Change

    TIHI

    
    index = []
    values = []

    for i, (p, interval) in enumerate(zip(probabilities, turnball_intervals)):
        if i == 0:
            index.append(interval.left)
            index.append(interval.right)
            values.append(1.0)
            values.append(1 - p)
            continue

        if interval.left != index[-1]:
            index.append(interval.left)
            values.append(values[-1])

        if interval.left == interval.right:
            values[-1] -= p
        else:
            index.append(interval.right)
            values.append(values[-1] - p)

    full_dataframe = pd.DataFrame(index=timeline, columns=[label + "_lower"])

    turnball_dataframe = pd.DataFrame(values, index=index, columns=[label + "_lower"])

    dataframe = full_dataframe.combine_first(turnball_dataframe).ffill().fillna(1)
    dataframe[label + "_upper"] = dataframe[label + "_lower"].shift(1).fillna(1)
    return dataframe

Instances