Merge pull request #46 from CosmoStat/fits

read fits file

Author: martinkilbinger

cs_util/cat.py

@@ -168,6 +168,39 @@ def write_fits_BinTable_file(
     hdu_list.writeto(output_path, overwrite=True)
 
 
+def read_fits_to_dict(file_path):
+    """Read Fits To Dict.
+
+    Read FITS file and return dictionary.
+
+    Parameters
+    ----------
+    file_path : str
+        input file path
+
+    Returns
+    -------
+    dict
+        file content
+
+    Raises
+    ------
+    IOError
+        if input file is not found
+    """
+    if not os.path.exists(file_path):
+        raise IOError(f"Input file '{file_path}' not found")
+
+    hdu_list = fits.open(file_path)
+    data_input = hdu_list[1].data
+    col_names = hdu_list[1].data.dtype.names
+    data = {}
+    for col_name in col_names:
+        data[col_name] = data_input[col_name]
+
+    return data
+
+
 def bin_edges2centers(bin_edges):
     """Bin Edges To Centers.
 

cs_util/plots.py

@@ -43,7 +43,8 @@ def savefig(fname, close_fig=True):
     fname : str
         output file name
     close_fig : bool, optional
-        closes figure if True (default)
+        closes figure if ``True`` (default); chose ``False``
+        to display figure in a jupyter notebook
 
     """
     plt.savefig(fname, facecolor="w", bbox_inches="tight")

pyproject.toml

@@ -6,8 +6,8 @@ authors = ["Martin Kilbinger <[email protected]>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = ">=3.9,<3.12"
-astropy = "5.2"
+python = ">=3.9,<3.13"
+astropy = "^5.0"
 camb = "1.5.4"
 datetime = "^5.5"
 numpy = "^1.26.4"