Don't import large datasets via .py files. It loads it all at once. Terrible memory use penalty.

Save the data in .npy or .npz or .h5 and load it from there. It can then load serially or incrementally. Better memory management.