Function format_matrix_str

eval_code/recons/utils/messages.py:63–82

(matrix)

        updated_data.to_csv(file_path, index=False)

def format_matrix_str(matrix):
    def format_float(num, total_width=20, decimal_places=12):
        # strip all right 0s, then strip '.'
        s = f"{num:{total_width}.{decimal_places}f}".rstrip("0").rstrip(".")
        # add space to the left
        return f"{s:>{total_width}}"
    formatted = [
        [
            # f"{num:20.12f}".rstrip("0").rstrip(".") if "." in f"{num}" else f"{num:20}" 
            # f"{num:20.12f}" if "." in f"{num}" else f"{num:20}" 
            format_float(num, total_width=15, decimal_places=8)
            for num in row
        ]
        for row in matrix
    ]
    rows = [
        f"        [{', '.join(num for num in row)}]"
        for row in formatted
    ]
    return "    [\n" + ",\n".join(rows) + "\n    ]"

def save_list_of_matrices(matrices_tosave: List[List[List[float]]], save_path: str) -> None:
    json_str = "[\n" + ",\n".join(format_matrix_str(mat) for mat in matrices_tosave) + "\n]"