一个简单的二维向量类

Posted 2020-11-04 五等码农

# _*_ coding: utf-8 _*_
__author__ = ‘pythonwu‘
__date__ = "2018/5/11 18:21"

from math import hypot

class Vecotr:
    def __init__(self,x=0,y=0):
        self.x = x
        self.y = y

    def __repr__(self):
        return "Vector(%r,%r)" %(self.x,self.y)

    def __abs__(self):
        return hypot(self.x,self.y)

    def __bool__(self):
        return bool(self.x or self.y)

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

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

注释：其实最简单是使用complex类来实现 ，但是当需要多维向量的时候就不是很方便了，这个类可以扩展至n维向量
注释：自定义对象bool方法（内置）__bool__方法不存在的时候，调用__len__判断真假