import data import numpy as np x = data.load() size=len(x) print(size) x_np = np.stack(x) x_np = np.expand_dims(x_np, axis=1) #y_np = np.stack(y) #y_np = np.expand_dims(y_np, axis=1) np.savez_compressed("data",x_np)