oasislmf.pytools.common.data¶
Attributes¶
Functions¶
|
load a file as a numpy ndarray |
|
load file as a single numpy array, |
|
Writes a custom dtype array with headers to csv with the provided row_fmt str |
|
Module Contents¶
- oasislmf.pytools.common.data.load_as_ndarray(dir_path, name, _dtype, must_exist=True, col_map=None)[source]¶
load a file as a numpy ndarray useful for multi-columns files Args:
dir_path: path to the directory where the binary or csv file is stored name: name of the file _dtype: np.dtype must_exist: raise FileNotFoundError if no file is present col_map: name re-mapping to change name of csv columns
- Returns:
numpy ndarray
- oasislmf.pytools.common.data.load_as_array(dir_path, name, _dtype, must_exist=True)[source]¶
- load file as a single numpy array,
useful for files with a binary version with only one type of value where their index correspond to an id. For example coverage.bin only contains tiv value for each coverage id coverage_id n correspond to index n-1
- Args:
dir_path: path to the directory where the binary or csv file is stored name: name of the file _dtype: numpy dtype of the required array must_exist: raise FileNotFoundError if no file is present
- Returns:
numpy array of dtype type
- oasislmf.pytools.common.data.write_ndarray_to_fmt_csv(output_file, data, headers, row_fmt)[source]¶
Writes a custom dtype array with headers to csv with the provided row_fmt str
This function is a faster replacement for np.savetxt as it formats each row one at a time before writing to csv. We create one large string, and formats all the data at once, and writes all the data at once.
WARNING: untested with string types in custom data.
- Args:
output_file (io.TextIOWrapper): CSV file data (ndarray[<custom dtype>]): Custom dtype ndarray with column names headers (list[str]): Column names for custom ndarray row_fmt (str): Format for each row in csv