Mysql
 sql >> Datenbank >  >> RDS >> Mysql

ERSETZEN Sie Zeilen in der MySQL-Datenbanktabelle mit Pandas DataFrame

Mit der Veröffentlichung von Pandas 0.24.0 gibt es jetzt einen offizieller Weg Um dies zu erreichen, übergeben Sie eine benutzerdefinierte Insert-Methode an to_sql Funktion.

Ich konnte das Verhalten von REPLACE INTO erreichen indem Sie dieses Callable an to_sql übergeben :

def mysql_replace_into(table, conn, keys, data_iter):
    from sqlalchemy.dialects.mysql import insert
    from sqlalchemy.ext.compiler import compiles
    from sqlalchemy.sql.expression import Insert

    @compiles(Insert)
    def replace_string(insert, compiler, **kw):
        s = compiler.visit_insert(insert, **kw)
        s = s.replace("INSERT INTO", "REPLACE INTO")
        return s

    data = [dict(zip(keys, row)) for row in data_iter]

    conn.execute(table.table.insert(replace_string=""), data)

Sie würden es so weitergeben:

df.to_sql(db, if_exists='append', method=mysql_replace_into)

Alternativ, wenn Sie das Verhalten von INSERT ... ON DUPLICATE KEY UPDATE ... wünschen stattdessen können Sie Folgendes verwenden:

def mysql_replace_into(table, conn, keys, data_iter):
    from sqlalchemy.dialects.mysql import insert

    data = [dict(zip(keys, row)) for row in data_iter]

    stmt = insert(table.table).values(data)
    update_stmt = stmt.on_duplicate_key_update(**dict(zip(stmt.inserted.keys(), 
                                               stmt.inserted.values())))

    conn.execute(update_stmt)

Dank an https://stackoverflow.com/a/11762400/1919794 für die Kompilierungsmethode.