def observe(v):
    v2 = (np.absolute(v) ** 2)
    c = np.random.choice(v.size, 1, p=v2)
    return c[0]
def observe(v):
    v2 = (np.absolute(v) ** 2)
    c = np.random.choice(v.size, 1, p=v2)
    return c[0]