1. import numpy as np
  3. def migrate_array(source_array, target_array, mapping, default_value=None):
  4. """
  5. Migrates data from a source array to a target array based on a mapping.
  6. Includes defensive checks to handle potential errors.
  8. Args:
  9. source_array (np.ndarray): The source array to migrate data from.
  10. target_array (np.ndarray): The target array to migrate data to.
  11. mapping (dict): A dictionary where keys are source array indices/values
  12. and values are target array indices/values.
  13. default_value (any, optional): The value to use if a mapping is missing. Defaults to None.
  15. Returns:
  16. np.ndarray: The target array with migrated data. Returns the original target_array
  17. if errors are encountered.
  19. Raises:
  20. TypeError: if source_array or target_array are not numpy arrays
  21. ValueError: if mapping is not a dictionary
  22. """
  24. if not isinstance(source_array, np.ndarray) or not isinstance(target_array, np.ndarray):
  25. raise TypeError("source_array and target_array must be numpy arrays.")
  27. if not isinstance(mapping, dict):
  28. raise ValueError("mapping must be a dictionary.")
  30. if source_array.shape != target_array.shape:
  31. print("Warning: Source and target arrays have different shapes. Returning original target_array.")
  32. return target_array
  34. for i in range(len(source_array)):
  35. try:
  36. target_index = mapping[source_array[i]]
  37. if target_index >= target_array.shape[0]:
  38. print(f"Warning: Mapping index {target_index} is out of bounds for target array. Using default value.")
  39. target_array[i] = default_value
  40. else:
  41. target_array[i] = target_array[target_index] # copy value
  42. except KeyError:
  43. if default_value is not None:
  44. target_array[i] = default_value
  45. print(f"Warning: No mapping found for value {source_array[i]}. Using default value.")
  46. else:
  47. print(f"Warning: No mapping found for value {source_array[i]}. Skipping.")
  48. pass # leave the value as is.
  49. except IndexError:
  50. print(f"Warning: Mapping index {target_index} is out of bounds for target array. Using default value.")
  51. target_array[i] = default_value
  52. return target_array

Add your comment