def plot(self, columns=None, loc=None, iloc=None, **kwargs):
        """"
        A wrapper around plotting. Matplotlib plot arguments can be passed in, plus:

        Parameters
        -----------
        columns: string or list-like, optional
          If not empty, plot a subset of columns from the ``cumulative_hazards_``. Default all.
        loc:

        iloc: slice, optional
          specify a location-based subsection of the curves to plot, ex:
                 ``.plot(iloc=slice(0,10))`` will plot the first 10 time points.
        """
        from matplotlib import pyplot as plt

        assert loc is None or iloc is None, "Cannot set both loc and iloc in call to .plot"

        def shaded_plot(ax, x, y, y_upper, y_lower, **kwargs):
            base_line, = ax.plot(x, y, drawstyle="steps-post", **kwargs)
            ax.fill_between(x, y_lower, y2=y_upper, alpha=0.25, color=base_line.get_color(), linewidth=1.0, step="post")

        def create_df_slicer(loc, iloc):
            get_method = "loc" if loc is not None else "iloc"

            if iloc is None and loc is None:
                user_submitted_ix = slice(0, None)
            else:
                user_submitted_ix = loc if loc is not None else iloc

            return lambda df: getattr(df, get_method)[user_submitted_ix]

        subset_df = create_df_slicer(loc, iloc)

        if not columns:
            columns = self.cumulative_hazards_.columns
        else:
            columns = _to_list(columns)

        set_kwargs_ax(kwargs)
        ax = kwargs.pop("ax")

        x = subset_df(self.cumulative_hazards_).index.values.astype(float)

        for column in columns:
            y = subset_df(self.cumulative_hazards_[column]).values
            index = subset_df(self.cumulative_hazards_[column]).index
            y_upper = subset_df(self.confidence_intervals_[column].loc["upper-bound"]).values
            y_lower = subset_df(self.confidence_intervals_[column].loc["lower-bound"]).values
            shaded_plot(ax, x, y, y_upper, y_lower, label=column, **kwargs)

        plt.hlines(0, index.min() - 1, index.max(), color="k", linestyles="--", alpha=0.5)

        ax.legend()
        return ax
def plot ( self , columns = None , loc = None , iloc = None , ** kwargs ) : from matplotlib import pyplot as plt assert loc is None or iloc is None , "Cannot set both loc and iloc in call to .plot" def shaded_plot ( ax , x , y , y_upper , y_lower , ** kwargs ) : base_line , = ax . plot ( x , y , drawstyle = "steps-post" , ** kwargs ) ax . fill_between ( x , y_lower , y2 = y_upper , alpha = 0.25 , color = base_line . get_color ( ) , linewidth = 1.0 , step = "post" ) def create_df_slicer ( loc , iloc ) : get_method = "loc" if loc is not None else "iloc" if iloc is None and loc is None : user_submitted_ix = slice ( 0 , None ) else : user_submitted_ix = loc if loc is not None else iloc return lambda df : getattr ( df , get_method ) [ user_submitted_ix ] subset_df = create_df_slicer ( loc , iloc ) if not columns : columns = self . cumulative_hazards_ . columns else : columns = _to_list ( columns ) set_kwargs_ax ( kwargs ) ax = kwargs . pop ( "ax" ) x = subset_df ( self . cumulative_hazards_ ) . index . values . astype ( float ) for column in columns : y = subset_df ( self . cumulative_hazards_ [ column ] ) . values index = subset_df ( self . cumulative_hazards_ [ column ] ) . index y_upper = subset_df ( self . confidence_intervals_ [ column ] . loc [ "upper-bound" ] ) . values y_lower = subset_df ( self . confidence_intervals_ [ column ] . loc [ "lower-bound" ] ) . values shaded_plot ( ax , x , y , y_upper , y_lower , label = column , ** kwargs ) plt . hlines ( 0 , index . min ( ) - 1 , index . max ( ) , color = "k" , linestyles = "--" , alpha = 0.5 ) ax . legend ( ) return ax
A wrapper around plotting. Matplotlib plot arguments can be passed in, plus:

        Parameters
        -----------
        columns: string or list-like, optional
          If not empty, plot a subset of columns from the ``cumulative_hazards_``. Default all.
        loc:

        iloc: slice, optional
          specify a location-based subsection of the curves to plot, ex:
                 ``.plot(iloc=slice(0,10))`` will plot the first 10 time points.