Read hdf5 snaps and concatenate

read_snaps.py

import numpy as np
import h5py
import glob

files = glob.glob('snap*')


coods = [None] * len(files)
ages = [None] * len(files)
metals = [None] * len(files)
mass = [None] * len(files)

for i,f in enumerate(files):

    hf = h5py.File(f,'r')

    coods[i] = hf['PartType4/Coordinates'][:]
    ages[i] = hf['PartType4/GFM_StellarFormationTime'][:]
    metals[i] = hf['PartType4/GFM_Metallicity'][:]
    mass[i] = hf['PartType4/GFM_InitialMass'][:]


coods = np.vstack(coods)
ages = np.hstack(ages)
metals = np.hstack(metals)
mass = np.hstack(mass)