0

我创建了一个类，其中主要的可交付数据块作为 pandas DataFrame 存储在一个属性中。我希望此类实例的默认显示行为与此 DataFrame 的相同。特别是在 iPython Notebook 中。

例如：

from pandas import DataFrame


class TestDFDisplay():
    def __init__(self):
        self.dataframe = DataFrame([[1, 2], [3, 4]])


tdf = TestDFDisplay()

当我：

tdf.dataframe

我得到一个 HTML 版本：

   0  1
0  1  2
1  3  4

当我：

tdf

我得到：

<__main__.TestDFDisplay instance at 0x000000001A836788>

我宁愿获得相同的 HTML：

   0  1
0  1  2
1  3  4

相反，我可以：

from pandas import DataFrame


class TestDFDisplay():
    def __init__(self):
        self.dataframe = DataFrame([[1, 2], [3, 4]])

    def __getattr__(self, item):
        try:
            return object.__getattribute__(self, item)
        except AttributeError:
            try:
                return getattr(self.dataframe, item)
            except:
                raise AttributeError


tdf = TestDFDisplay()

但这是一种非常笨拙的方法，它将任何从类实例获取属性的尝试转移到尝试从 DataFrame 中获取属性。这可行，但我宁愿更精确，并执行以下操作：

from pandas import DataFrame


class TestDFDisplay():
    def __init__(self):
        self.dataframe = DataFrame([[1, 2], [3, 4]])

    def __repr__(self):
        return self.dataframe.__repr__()


tdf = TestDFDisplay()

所以当我：

tdf

我得到的文本版本（与此处显示的相同）而不是我想要的 HTML 版本：

   0  1
0  1  2
1  3  4

没关系。这只是意味着“ repr ”方法不是在 DataFrame 上调用以在 iPython Notebook 中显示 HTML 的正确方法。

我的问题是：我应该在 DataFrame 处重定向的正确方法是什么？

4

1 回答 1

4

您想要_repr_html_IPython/Jupyter在使用富 (HTML) 显示系统时寻找的东西。

所以在你的课上

class TestDFDisplay():
    def __init__(self):
        self.dataframe = DataFrame([[1, 2], [3, 4]])

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

应该管用。

于 2015-09-29T18:27:27.963 回答