def load_data(datapath):
datapath = verify_path(datapath)
if str(datapath).split('.')[-1] == 'parquet':
data = pd.read_parquet(datapath)
elif str(datapath).split('.')[-1] == 'hdf5':
data = pd.read_hdf5(datapath)
return data
def load_data( datapath, file_format=None ):
datapath = verify_path( datapath )
if file_format is None:
file_format = str(datapath).split('.')[-1]
if file_format=='parquet':
data = pd.read_parquet( datapath )
elif file_format=='hdf5':
data = pd.read_hdf5( datapath )
elif file_format=='csv':
data = pd.read_csv( datapath )
else:
try:
data = pd.read_csv( datapath )
except:
print('Cannot load file', datapath)
return data