1. import numpy as np
  3. def convert_array(input_array, target_type='float64', min_value=None, max_value=None):
  4. """
  5. Converts an input array to a specified data type and performs sanity checks.
  7. Args:
  8. input_array (array-like): The array to convert. Can be a list, tuple, or numpy array.
  9. target_type (str): The desired data type (e.g., 'float64', 'int32', 'float32').
  10. min_value (float, optional): Minimum acceptable value for float types. Defaults to None.
  11. max_value (float, optional): Maximum acceptable value for float types. Defaults to None.
  13. Returns:
  14. numpy.ndarray: The converted numpy array, or None if conversion fails or sanity checks fail.
  16. Raises:
  17. TypeError: If input is not array-like.
  18. ValueError: If target_type is invalid.
  19. """
  20. try:
  21. # Convert to numpy array for easier handling
  22. arr = np.array(input_array)
  24. # Convert to the target data type
  25. if target_type == 'float64':
  26. arr = arr.astype(np.float64)
  27. elif target_type == 'float32':
  28. arr = arr.astype(np.float32)
  29. elif target_type == 'int32':
  30. arr = arr.astype(np.int32)
  31. elif target_type == 'int64':
  32. arr = arr.astype(np.int64)
  33. else:
  34. raise ValueError("Invalid target_type. Choose from 'float64', 'float32', 'int32', 'int64'.")
  36. # Sanity checks for float types
  37. if np.issubdtype(arr.dtype, np.floating):
  38. if min_value is not None and np.any(arr < min_value):
  39. print(f"Sanity check failed: Value below minimum ({min_value}) found.")
  40. return None
  41. if max_value is not None and np.any(arr > max_value):
  42. print(f"Sanity check failed: Value above maximum ({max_value}) found.")
  43. return None
  45. return arr
  47. except TypeError:
  48. print("Error: Input must be array-like.")
  49. return None
  50. except ValueError as e:
  51. print(f"Error: {e}")
  52. return None
  53. except Exception as e:
  54. print(f"An unexpected error occurred: {e}")
  55. return None
  58. if __name__ == '__main__':
  59. # Example Usage
  60. data = [1, 2, 3.5, 4, 5]
  61. converted_array = convert_array(data, target_type='float64', min_value=2.0, max_value=4.0)
  63. if converted_array is not None:
  64. print("Converted array:", converted_array)
  66. data2 = [1, 2, 3, 4, 5]
  67. converted_array2 = convert_array(data2, target_type='int32')
  68. if converted_array2 is not None:
  69. print("Converted array:", converted_array2)
  71. data3 = [1, 2, 3, 4, 5]
  72. converted_array3 = convert_array(data3, target_type='invalid_type')
  73. if converted_array3 is not None:
  74. print("Converted array:", converted_array3)

Add your comment