1. import numpy as np
  3. def restore_arrays(arrays, default_value=0):
  4. """
  5. Restores data of arrays for diagnostics with fallback logic.
  7. Args:
  8. arrays (list of numpy arrays): A list of numpy arrays to restore.
  9. default_value (any, optional): The default value to use if data is missing. Defaults to 0.
  11. Returns:
  12. list of numpy arrays: A list of numpy arrays with restored data.
  13. """
  14. restored_arrays = []
  15. for arr in arrays:
  16. if isinstance(arr, np.ndarray): # Check if it's a numpy array
  17. if np.isnan(arr).any(): # Check if array contains NaN values
  18. # Fallback logic: Replace NaN values with default_value
  19. arr[np.isnan(arr)] = default_value
  20. restored_arrays.append(arr)
  21. else:
  22. #If not a numpy array, just return the original value
  23. restored_arrays.append(arr) # Preserve non-array elements
  24. return restored_arrays

Add your comment