Error when setting labels=False in plotting.add_at_risk_counts().

For lifelines 0.25.0. Getting an AttributeError: 'NoneType' object has no attribute 'replace' because all labels are set to None if labels is False:

    if labels is None:
        labels = [f._label for f in fitters]
    elif labels is False:
        labels = [None] * len(fitters)

    labels = [l.replace("_", r"\_") for l in labels]

And is it possible to disable the rows ‘Censored’ and/or ‘Events’ in the ‘At risk’ counts? To have that configurable, or is it already configurable?!

[CamDavidsonPilon]

1. I’ve add a rows_to_show kwarg in add_at_risk_counts that can configure what is shown.
2. I think I’ve fixed the alignment problem (by remove the italics…)
3. the original labels=False problem should be fixed.

All in the latest v0.25.1

[agnesbao]

I think it’s the italic font somehow creates extra space on the first column. If i take out the italic font setting, or add it to the rest of ticks by changing this line: https://github.com/CamDavidsonPilon/lifelines/blob/cae49555be8c66a09c95ba28f52189941aa89821/lifelines/plotting.py#L491 into lbl += "$\\mathit{}$" + s.format(c) then it look aligned. (Haven’t tried the is_latex_enabled() yet)