Как взять строку "str" ​​и понять, что это тип данных: str

Question

У меня есть файл Yaml, в котором я обозначаю столбцы и типы данных CSV. Я буду использовать это, чтобы прочитать фрейм данных. Файл имеет формат:

columns:
    company: str
    country: str
    address: str

Когда я читаю файл yaml, я получаю:

{
    "columns": {
        { "company": "str"},
        { "country": "str"},
        { "address": "str"}
    }
}

Но мне нужно это как:

    "columns": {
        { "company": str},
        { "country": str},
        { "address": str}
    }
}

На данный момент мой хак - использовать справочную таблицу, которая конвертируется в тип.

{ 
    "int": int, 
    "str": str
}

Answer 1

Вы можете import builtins и получить типы в качестве атрибутов этого модуля:

import builtins

config['columns'] = {k: getattr(builtins, v) for k, v in config['columns'].items()}