当对象在列表或字典中时__str__方法不起作用

时间:2017-09-25 13:13:33

标签: python class

在此示例中,第一个print语句输出ball.__str__()返回的字符串,而其他两个不输出:

class Ball:

    def __init__(self, parent, xpos = 50, ypos = 50, radius = 100, vx = 0, vy = 0, mass = 1):
        """
        x,y are positions
        vx and vy are velocities in cells/second
        """
        self.x = xpos
        self.y = ypos
        self.r = radius
        self.vx = vx
        self.vy = vy
        self.mass = mass
        self.board = parent

    def __str__(self):
        return "Ball: x={0}, y={1}, r={2}, vx={3}, vy={4}".format(self.x,self.y,self.r,self.vx,self.vy)

class Board:

    def __init__(self, width = 100, height = 100, sps = 2):
        pass

board = Board()

ball = Ball(board)
ball_list = [Ball(board), Ball(board)]
ball_dict = {'ball_1':Ball(board), 'ball_2':Ball(board)}

print(ball)
print(ball_list)
print(ball_dict)

输出:

Ball: x=50, y=50, r=100, vx=0, vy=0
[<__main__.Ball object at 0x106f79f98>, <__main__.Ball object at 0x106f79fd0>]
{'ball_1': <__main__.Ball object at 0x106f81048>, 'ball_2': <__main__.Ball object at 0x106f81080>}

问题:

  1. 为什么python会以这种方式运行?
  2. 如何让 str 字符串出现在列表和字典中?

2 个答案:

答案 0 :(得分:6)

print使用__str__方法,但分别打印dictlist调用dict.__str__ / list.__str__,使用{__repr__ 1}}序列化包含项的方法。在您的课程中定义__repr__以模仿__str__。例如。这样做:

class Ball:
    ...

    def __str__(self):
        ...

    __repr__ = __str__

请注意,__repr__ 应该返回一个表示,该表示最好是有效的Python代码,例如Ball(Board(100, 100, 2))

答案 1 :(得分:1)

一个例子应该有所帮助。

class Foo():
     def __str__(self): 
         return '__str__'

     def __repr__(self): 
         return '__repr__'

x = Foo(); print(x)
__str__

lst = [Foo(), Foo(), Foo()]    
print(lst)
[__repr__, __repr__, __repr__]

在数据结构中,调用__repr__方法,而不是__str__。如果你没有定义这样的方法,python会回到__repr__提供的默认object

如上所述,修复方法是定义__repr__方法,并让它引用您班级当前定义的__str__方法。

你也可以在类定义之后 monkeypatch 方法,如下所示:

Foo.__repr__ = Foo.__str__