datatype.py 985 B

12345678910111213141516171819202122232425
  1. # Copyright 2023 Avaiga Private Limited
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
  4. # the License. You may obtain a copy of the License at
  5. #
  6. # http://www.apache.org/licenses/LICENSE-2.0
  7. #
  8. # Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
  9. # an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
  10. # specific language governing permissions and limitations under the License.
  11. import re
  12. import pandas as pd
  13. def _get_data_type(value):
  14. if not isinstance(value, str):
  15. if pd.api.types.is_bool_dtype(value):
  16. return "bool"
  17. elif pd.api.types.is_integer_dtype(value):
  18. return "int"
  19. elif pd.api.types.is_float_dtype(value):
  20. return "float"
  21. return re.match(r"^<class '(.*\.)?(.*?)(\d\d)?'>", str(type(value))).group(2)