如何使用Postgres中CSV文件中的值更新所选行？

Question

我正在使用Postgres，并想进行一个大型更新查询，该查询将从CSV文件中提取，可以说我有一个表，该表具有(id, banana, apple)。

我想运行一个更新来更改香蕉而不是苹果，每个新的香蕉及其ID都将保存在CSV文件中。

我尝试查看Postgres网站，但这些示例使我丧命。

Answer 1

您可以尝试下面用python编写的代码，输入文件是要更新到表中的csv文件。每行都基于逗号进行拆分，因此对于每行，row [0]是第一列下的值，row [1]是第二列下的值，依此类推。    import csv    import xlrd    import os    import psycopg2    import django    from yourapp import settings    django.setup()    from yourapp import models    try:       conn = psycopg2.connect("host=localhost dbname=prodmealsdb        user=postgres password=blank")       cur = conn.cursor()       filepath = '/path/to/your/data_to_be_updated.csv'       ext = os.path.splitext(filepath)[-1].lower()       if (ext == '.csv'):           with open(filepath) as csvfile:          next(csvfile)          readCSV = csv.reader(csvfile, delimiter=',')          for row in readCSV:              print(row[3],row[5])              cur.execute("UPDATE your_table SET column_to_be_updated = %s where               id = %s", (row[5], row[3]))              conn.commit()          conn.close()          cur.close()    except (Exception, psycopg2.DatabaseError) as error:    print(error)    finally:    if conn is not None:      conn.close()