코딩 더 매트릭스 - 3장 벡터 (2)

def square(x):
    """Return the square of x.

    >>> square(2)
    4
    >>> square(-2)
    4
    """
    return x * x

class Vec:
    def __init__(self, labels, function):
        self.D = labels
        self.f = function

    def __repr__(self):
		"""
		Vec({...}, {...}) 인스턴스 혹은 인스턴스를 담은 변수를 호출했을때 문자열로 반환하는 값
		e.g.
		>>> v = Vec({...}, {...})
		>>> v
		Vec({...}, {...})
		"""
        return 'Vec(%s,%s)' % (self.D, self.f)

    def __str__(self):
		"""
		print 함수로 인스턴스 혹은 인스턴스를 담은 변수를 호출했을때 문자열로 반환하는 값
		아래 코드의 경우 다음과 같이 출력합니다.
		>>> v = Vec({1,2}, {1: 'a', 2: 'b'})
		1  2
		----
		a  b
		"""
        pad = 2
        max_len = max({ len(str(v)) for v in self.D } | { len(str(v)) for v in self.f.values() }) + pad
        sorted_D = sorted(self.D)
        key_line = ''.join([ str(d).rjust(max_len) for d in sorted_D ])[pad:]
        dash_line = ''.join([ '-' * max_len for i in range(len(sorted_D)) ])[pad:]
        value_line = ''.join([ str(self[d]).rjust(max_len) for d in sorted_D ])[pad:]
        return '%s\n%s\n%s' % (key_line, dash_line, value_line)

    def __getitem__(self, key):
		"""
		키값으로 값을 얻는 연산자 "[ ]" 정의
		"""
        return self.f[key] if key in self.f else 0

    def __setitem__(self, key, value):
		"""
		키값으로 값을 할당하는 연산 정의
		e.g. v[key] = value
		"""
        if key not in self.D:
            raise ValueError("key is not in D")
        self.f[key] = value

	def __add__(self, other):
		"""
		덧셈 연산자 "+" 정의
		"""
        return Vec(self.D, { d: self[d] + other[d] for d in self.D if self[d] or other[d] })

    def __neg__(self):
		"""
		음의 연산자 "-" 정의
		e.g. -v
		"""
        return Vec(self.D, { d: -self[d] for d in self.D })

    def __sub__(self, other):
		"""
		뺄셈 연산자 "-" 정의
		"""
        return self + (-other)

    def __mul__(self, other):
		"""
		곱셈 연산자 "*" 정의
		"""
        if isinstance(other, Vec):
            return sum([ self[d] * other[d] for d in self.D ])
        else:
            return Vec(self.D, { d: self[d] * other for d in self.D })
	"""
	mul : 매개변수 self는 "*" 의 왼쪽객체, other는 "*" 의 오른쪽객체
	rmul : 매개변수 self는 "*" 의 오른쪽객체, other는 "*" 의 왼쪽객체
	"""
    __rmul__ = __mul__

    def __floordiv__(self, other):
		"""
		나눗셈 연산자 "/" 정의
		"""
        return Vec(self.D, { d: self[d] / other for d in self.D })
    __truediv__ = __floordiv__

    def __eq__(self, other):
		"""
		일치 연산자 "==" 정의
		"""
        D = set(self.f.keys()) | set(other.f.keys())
        for d in D:
            if self[d] != other[d]:
                return False
        return True

	def copy(self):
		"""
		같은 주소값을 가리키는 객체가 아닌 내용은 같지만 완전히 다른 객체 생성(복사)
		"""
        return Vec(self.D.copy(), self.f.copy())

# 리스트를 벡터로
def list2vec(L):
    return Vec(set(range(len(L))), {k:L[k] for k in range(len(L))})

# 영벡터
def zero_vec(D):
    return Vec(D, {})