Dispaly DataFrame in Jupyter Notebook along with other attributes

Question

If I just construct a pandas DataFrame in a Jupyter notebook, then the output looks nice:

frame = pd.DataFrame({'a': [1,2,3]})
frame

However, if I make my own class whose repr is a pandas.DataFrame, then that formatting is lost:

class Foo:
    def __init__(self, frame, bar):
        self.frame = frame
        self.bar = bar
    
    def __repr__(self):
        return repr(self.frame)

How can I define the repr of my class Foo, such that when I display it in a Jupyter Notebook, it'll look just like it would if it was a pandas.DataFrame?

did you see this answer https://stackoverflow.com/questions/26873127/show-dataframe-as-table-in-ipython-notebook — Atanas Atanasov, Jan 17 '23 at 11:10
yes but I can't just return `HTML(df2.to_html())` from `__repr__`, as `__repr__` needs to return a string — ignoring_gravity, Jan 17 '23 at 11:22

Wayne · Answer 1 · 2023-01-17T22:16:51.170

The key resource you seek for guidance in making your class integrate well with Jupyter is covered under _repr_mimebundle_ in the 'Integrating your objects with IPython¶' section in the documentation.

I'm trying to look at where Pandas encodes it so that you can match but not finding quickly right now. (Some of it seems here but it's more complex than the latex example I point to below because of css and more.)

See another example of use of _repr_mimebundle_ here.

Dispaly DataFrame in Jupyter Notebook along with other attributes

1 Answers1