PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Comment mettre à jour efficacement une colonne dans une grande table PostgreSQL en utilisant Python/psycopg2 ?

unnest les données pour tout faire d'un coup :

def create_random(i):
    return random() * i

curs.execute("select gid, number from t;")
data = curs.fetchall()

results = []
for i in data:
    results.append((create_random(i[1]), i[0]))

curs.execute("""
    update t
    set
        rnd = s.rnd,
        result = number * s.rnd
    from unnest(%s) s(rnd numeric, gid integer)
    where t.gid = s.gid;
""", (results,))

con.commit()

Tableau t :

create table t (
    gid integer,
    number integer,
    rnd float,
    result float
);