orthogonal_mod

orthogonal_mod.py

import numpy as np

#pythran export orthogonal_mod (int[:], int)
#pythran export orthogonal_mod (int32[:], int)
def orthogonal_mod (x, nbits):
    out = np.zeros ((len(x), 1<<nbits), dtype=complex)
    for e in range (len (x)):
        out[e,x[e]] = 1
    return out

def test1 ():
    x = np.arange (16)
    u = orthogonal_mod (x, 4)
    return u

test_orth_mod.py

import numpy as np
from orthogonal_mod import orthogonal_mod

x = np.array ([0,1,0,1], dtype=np.int32)
o = orthogonal_mod (x, 1)