使用整数值类型从 CSV 文件插入 PostgreSQL 表

Question

  I want to insert data from a CSV file into a PostgreSQL table. The table 

结构如下。但我无法输入 INTEGER 类型的值。 它显示错误，例如- DataError：整数的无效输入语法：“vendor_phone” 第 1 行：...vendor_phone,vendor_address)VALUES ('vendor_name','vendor_ph...

如果我使用 VARCHAR 类型，它工作正常。但我需要使用整数值。

   CREATE TABLE vendors (
        vendor_id SERIAL PRIMARY KEY,
        vendor_name VARCHAR(100) NOT NULL,
        vendor_phone INTEGER,
        vendor_address VARCHAR(255) NOT NULL 
    )

---

    import psycopg2
    import csv

    database = psycopg2.connect (database = "supplier", user="postgres", password="1234", host="localhost", port="5432")

cursor = database.cursor()

    vendor_data = csv.reader(open('vendors.csv'),delimiter=',')

    for row in vendor_data:

        cursor.execute("INSERT INTO vendors (vendor_name,vendor_phone,vendor_address)"\
            "VALUES (%s,%s,%s)",
           row)

    print("CSV data imported")

    cursor.close()
    database.commit()
    database.close()

Answer 1

您可以使用以下语句代替光标，将数据直接从 CSV 加载到跳过 CSV 文件标题的表中

COPY vendors (vendor_name,vendor_phone,vendor_address) FROM 'vendors.csv' CSV HEADER;