在postgreql中使用COPY时，psycopg2.errors.InvalidTextPresation

Question

我正在使用一个自定义的可调用的熊猫。to_sql（）。下面的代码片段来自pandas文档，用于使用它

import csv
from io import StringIO

def psql_insert_copy(table, conn, keys, data_iter):
    """
    Execute SQL statement inserting data

    Parameters
    ----------
    table : pandas.io.sql.SQLTable
    conn : sqlalchemy.engine.Engine or sqlalchemy.engine.Connection
    keys : list of str
        Column names
    data_iter : Iterable that iterates the values to be inserted
    """
    # gets a DBAPI connection that can provide a cursor
    dbapi_conn = conn.connection
    with dbapi_conn.cursor() as cur:
        s_buf = StringIO()
        writer = csv.writer(s_buf)
        writer.writerows(data_iter)
        s_buf.seek(0)

        columns = ', '.join('"{}"'.format(k) for k in keys)
        if table.schema:
            table_name = '{}.{}'.format(table.schema, table.name)
        else:
            table_name = table.name

        sql = 'COPY {} ({}) FROM STDIN WITH CSV'.format(
            table_name, columns)
        cur.copy_expert(sql=sql, file=s_buf)

但是在使用这个复制功能时，我得到了错误

psycopg2.errors.InvalidTextRepresentation: invalid input syntax for integer: "3.0"

当我习惯于使用_sql（）函数而不使用自定义可调用的“psql_insert_copy（）”时，作为此表模式和值的输入在最初工作时不会出现问题。我正在使用sqlalchemy引擎获取连接光标

Answer 1

我建议在表中使用字符串字段进行此类操作，或者手动编写整个（sql）脚本，指示表字段的类型