python - Pandas 数据框在 Jupyter 中呈现为表格

Question

如果我有一个pandas.DataFrame df，只需做

df

在一个空的 Jupyter notebook 单元格中，它会很好地呈现为表格。

有了一个环绕数据框的类，我如何才能以相同的方式呈现底层数据框？

class Wrapper:
  def __init__(self, df):
    self._df = df

  def __repr__(self):
    ## to_html does not work
    return self._df.to_html()

这给出了想法，但 to_html 似乎无法做到这一点。我能做什么？

Answer 1

如果我正确理解了这个问题，您希望在调用包装器对象时将 df 打印为表格吗？如果是这种情况，那么您可以包装数据框的 _repr_html_ 方法。

class Wrapper:
    def __init__(self, df):
        self._df = df

    def _repr_html_(self):
        return self._df._repr_html_()

或者:

class Wrapper:
    def __init__(self, df):
        self._df = df

    self.repr_html = self._df._repr_html_

创建对象后:

mytable = Wrapper(df)

您可以通过调用 mytable 在笔记本中将 df 显示为表格

mytable # displays the df