Staging
v0.8.1
Revision f78dd3452d99318195bf3dc7a4266ca5f12dfe11 authored by Georg Brandl on 13 August 2009, 08:58:24 UTC, committed by Georg Brandl on 13 August 2009, 08:58:24 UTC
svn+ssh://svn.python.org/python/branches/py3k

........
  r73592 | ezio.melotti | 2009-06-28 00:58:15 +0200 (So, 28 Jun 2009) | 1 line

  Updated the last example as requested in #6350
........
  r73823 | ezio.melotti | 2009-07-04 03:14:30 +0200 (Sa, 04 Jul 2009) | 1 line

  #6398 typo: versio. -> version.
........
1 parent a5ea7e3
Raw File
Vec.py
# A simple vector class


def vec(*v):
    return Vec(*v)


class Vec:

    def __init__(self, *v):
        self.v = list(v)

    def fromlist(self, v):
        if not isinstance(v, list):
            raise TypeError
        self.v = v[:]
        return self

    def __repr__(self):
        return 'vec(' + repr(self.v)[1:-1] + ')'

    def __len__(self):
        return len(self.v)

    def __getitem__(self, i):
        return self.v[i]

    def __add__(self, other):
        # Element-wise addition
        v = list(map(lambda x, y: x+y, self, other))
        return Vec().fromlist(v)

    def __sub__(self, other):
        # Element-wise subtraction
        v = list(map(lambda x, y: x-y, self, other))
        return Vec().fromlist(v)

    def __mul__(self, scalar):
        # Multiply by scalar
        v = [x*scalar for x in self.v]
        return Vec().fromlist(v)



def test():
    a = vec(1, 2, 3)
    b = vec(3, 2, 1)
    print(a)
    print(b)
    print(a+b)
    print(a-b)
    print(a*3.0)

test()
back to top