merge_npy_files.py

import numpy as np
import os

path = ''
files = os.listdir(path)
data = np.zeros((1, 128))
identifiers = np.full((data.shape[0]), 0)

for file in files:
    temp = np.load(path+file)
    data = np.concatenate((data, temp), axis=0)
    identifiers = np.append(identifiers, np.full((temp.shape[0]), int(file.split('.')[0][-3:]) ))
    print('Done', file, data.shape[0])

np.savez('merged_embeddings.npz', filelist=identifiers, features=data)