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