Вставить тип данных enum через pandas.to_sql

У меня есть фреймворк pandas следующим образом:

In [93]: df = pd.DataFrame([[1, 'a'], [2, 'b'], [3, 'a']], columns=['val', 'types'])

In [94]: df
Out[94]:
   val types
0    1    a
1    2    b
2    3    a

In [95]: df.dtypes
Out[95]:
val       int64
types    object
dtype: object

# convert the 'types' column to category
In [96]: df.types = pd.Categorical(df.types)

Предполагая, что у меня уже есть таблица postgres со столбцом types как ENUM, как мне вставить данные в мою БД через типы sqlachemy с методом pandas to_sql?

Я пробовал это, но не смог заставить его работать:

In [101]: class myEnum(enum.Enum):
     ...:     a = 1
     ...:     b = 2
     ...:

In [102]: from sqlalchemy.types import *

In [103]: sql_dtypes = {'val' : SmallInteger(), 'types': Enum(myEnum)}

In [104]: df.to_sql('tablename', conn, if_exists='replace', index=False,
                    dtype=sql_dtypes)

работает, что дает мне следующую ошибку:

sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) type "myEnum" already 
exists

[SQL: "CREATE TYPE myEnum AS ENUM ('a', 'b')"] (Background on this 
error at: http://sqlalche.me/e/f405)

Выбор опции schema из pd.io.json.build_table_schema тоже не помог.

почему не if_exists = "append"? чтобы панды не воссоздали стол для вас

— 17.05.2020 07:02

python postgresql pandas sqlalchemy

18.09.2018 20:59

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

668

Ответы 1

Обходной путь, который я нашел на данный момент из здесь, - это сначала записать тип данных как varchar, а затем преобразовать его в enum.

In [104]: df.to_sql('tablename', conn, if_exists='replace', index=False,
                    dtype=sql_dtypes)

In [105]: conn.execute("""ALTER TABLE tablename 
                          ALTER COLUMN types TYPE myEnum using types::myEnum;""")

Но я надеюсь, что есть лучшее решение.

19.09.2018 09:32