pandas_astype¶

pandas_astype(columns, postgres_dtype)¶

Параметры:

columns (list[str]) – Список имен колонок
postgres_dtype (list[PostgreSQLDtype]) – Список типов данных PostgreSQL

Результат:

Словарь соответствий для pandas dtype

Тип результата:

dict[str, str]

Генерация типов данных pandas из типов PostgreSQL.

Описание:

Преобразует типы данных PostgreSQL в соответствующие типы pandas для корректного создания DataFrame. Используется при конвертации данных из PGPack формата в pandas DataFrame.

Словарь соответствий PANDAS_TYPE:

Соответствие типов PostgreSQL → pandas dtype¶
Тип Python	Тип pandas
`NoneType`	`"nan"`
`bool`	`"?"`
`date`	`"datetime64[ns]"`
`float`	`"float64"`
`str`	`"string"`

Возвращает:

Словарь, где ключ - имя колонки, значение - pandas dtype или None если соответствие не найдено.

Пример использования:

columns = ["id", "name", "price"]
pg_types = [PostgreSQLDtype.Int4, PostgreSQLDtype.Text, PostgreSQLDtype.Numeric]

astype_dict = pandas_astype(columns, pg_types)
# {'id': None, 'name': 'string', 'price': None}

df = pd.DataFrame(data).astype(astype_dict)

Примечание:

Для типов, не включенных в словарь PANDAS_TYPE, возвращается None.