Mysql
 sql >> Base de Dados >  >> RDS >> Mysql

Como inserir um dataframe de pandas enorme na tabela MySQL com instrução de inserção paralela?


Você pode fazer algumas coisas para conseguir isso.

Uma maneira é usar um argumento adicional ao escrever no sql.
df.to_sql(method = 'multi')

De acordo com esta documentação , passar 'multi' para o argumento do método permite inserir em massa.

Outra solução é construir uma função de inserção personalizada usando multiprocessing.dummy.aqui está o link para a documentação:https://docs.python.org/2/library/multiprocessing.html#module-multiprocessing.dummy
import math
from multiprocessing.dummy import Pool as ThreadPool

...

def insert_df(df, *args, **kwargs):
    nworkers = 4 # number of workers that executes insert in parallel fashion

    chunk = math.floor(df.shape[0] / nworkers) # number of chunks
    chunks = [(chunk * i, (chunk * i) + chunk) for i in range(nworkers)]
    chunks.append((chunk * nworkers, df.shape[0]))
    pool = ThreadPool(nworkers)

    def worker(chunk):
        i, j = chunk
        df.iloc[i:j, :].to_sql(*args, **kwargs)

    pool.map(worker, chunks)
    pool.close()
    pool.join()

....

insert_df(df, "foo_bar", engine, if_exists='append')

O segundo método foi sugerido em https://stackoverflow.com/a/42164138/5614132 .