如何在ipython控制台中定义类输出

时间:2019-03-04 01:26:23

标签: python python-3.x class ipython

假设我有以下代码:

class PrintStuff:
    def __init__(self,stuff):
        self._stuff = stuff

    def __str__(self):
        return self._stuff

如果我在ipython中创建它的一个实例并键入该实例的名称,它将打印出该类的名称及其在内存中的位置。但是,如果我将实例放入打印功能,它将按预期打印:

pstuff = PrintStuff('print this stuff')

pstuff
Out[44]: <__main__.PrintStuff at 0x7fae0531de80>

print(pstuff)
print this stuff

我该如何做才能使ipython控制台打印出与打印功能相同的内容?例如,一个熊猫系列具有我正在寻找的行为类型:

series = pd.Series({'x':[1,2],'y':[2,3],'z':[3,4]})

series
Out[47]: 
x    [1, 2]
y    [2, 3]
z    [3, 4]
dtype: object

print(series)
x    [1, 2]
y    [2, 3]
z    [3, 4]
dtype: object

1 个答案:

答案 0 :(得分:3)

您需要添加一个__repr__

默认情况下,对象__repr__返回类型和内存地址,这是您在控制台输出(例如)时看到的内容:

<__main__.PrintStuff at 0x7fae0531de80>

__repr__重载之后,您可以控制输出是什么。

class PrintStuff:
    def __init__(self,stuff):
        self._stuff = stuff

    def __str__(self):
        return self._stuff

    def __repr__(self):
        return str(self)


pstuff = PrintStuff('print this stuff')
pstuff

输出:

print this stuff