实现一个简单二维向量类

#!/usr/bin/env python
# coding=utf-8

from math import hypot


class Vector:

    def __init__(self, x=0, y=0):
        self.x = x
        self.y = y
    def __repr__(self):
        """
        内置函数repr, 把一个对象用字符串的形式表达出来以便辨认,这就是'字符串表示形式'.
        repr就是通过__repr__这个特殊方法来得到一个对象的字符串表示形式的.
        如果没有实现__repr__,会得到<Vector object at 0x10e100070>

        __repr__ 与 __str__的区别, 前者方便我们调试和记录日志, 后者给终端用户看.

        如果要二选一, __repr__是更好的选择, 因为如果一个对象没有__str__函数,
        Python在需要调用是会用__repr__作为替代.
        """
        return 'Vector(%r, %r)' % (self.x, self.y)
    def __abs__(self):
        return hypot(self.x, self.y)

    def __bool__(self):
        return bool(abs(self))

    def __add__(self, other):
        x = self.x + other.x
        y = self.y + other.y
        return Vector(x, y)

    def __mul__(self, scalar):
        return Vector(self.x * scalar, self.y * scalar)

v1 = Vector(1, 4)
v2 = Vector(2, 8)
print(v1 + v2)

EngineerLeo
598 声望38 粉丝

专注于云原生、AI等相关技术