Convex combination for Coursera's "Coding the Matrix" course

convex.py

import time
from vec import Vec
from plotting import plot

def list2vec(l):
    D = set(range(len(l)))
    return Vec(D, {k: v for k, v in enumerate(l)})

def draw_line(p1, p2):
    resolution = 10
    alphas = (float(i)/resolution for i in range(resolution+1))
    scalers = ((alpha, 1-alpha) for alpha in alphas)
    points = (alpha*p1+beta*p2 for alpha, beta in scalers)
    points = [list(x.f.values()) for x in points]
    plot(points, 5)
    time.sleep(10)

u = list2vec([2, 1])
v = list2vec([-2, 2])

draw_line(u, v)