import numpy as np

sequences = [[1, 2, 3, 1, 4, 5], [7, 8, 9, 2, 1, 4, 5], [7, 8, 9, 2, 1, 4]]
embeddings = np.random.random((10,5))


##Rozšírte matice v tak aby každá mala rovnaký
##rozmer podľa maximálnej dĺžky zoznamu doplnením 0 hodnôt a spojte ich do výslednej matice data
## o rozmeroch počet sekvencií x max. dĺžka sekvencie x 5

zoznam = [];
biggestMatrix = 0;
for s in sequences:
    v = np.random.random((len(s),5))
    for i,a in enumerate(s):
       v[i] = embeddings[a]
    if biggestMatrix < v.shape[0]:
         biggestMatrix = v.shape[0]
    zoznam.append(v)

for index,matica in enumerate(zoznam):
    if (matica.shape[0] < biggestMatrix) : zoznam[index] = np.concatenate((zoznam[index], np.zeros((biggestMatrix - matica.shape[0];, 5))))

for index,matica in enumerate(zoznam):
    if matica.shape[0] < biggestMatrix:
        rozdiel = biggestMatrix - matica.shape[0];
        rozsirovac = np.zeros((rozdiel, 5))
        zoznam[index] = np.concatenate((zoznam[index], rozsirovac))

zoznam