📜  Python PostgreSQL-插入数据

📅  最后修改于: 2020-11-07 07:53:53             🧑  作者: Mango


您可以使用INSERT INTO语句将记录插入到PostgreSQL中的现有表中。执行此操作时,需要指定表的名称以及表中各列的值。

句法

以下是INSERT语句的建议语法-

INSERT INTO TABLE_NAME (column1, column2, column3,...columnN)
VALUES (value1, value2, value3,...valueN);

其中,column1,column2,column3,..是表中各列的名称,而value1,value2,value3,…是您需要插入表中的值。

假设我们使用CREATE TABLE语句创建了一个名称为CRICKETERS的表,如下所示-

postgres=# CREATE TABLE CRICKETERS (
   First_Name VARCHAR(255),
   Last_Name VARCHAR(255),
   Age INT,
   Place_Of_Birth VARCHAR(255),
   Country VARCHAR(255)
);
CREATE TABLE
postgres=#

以下PostgreSQL语句在上面创建的表中插入一行-

postgres=# insert into CRICKETERS 
   (First_Name, Last_Name, Age, Place_Of_Birth, Country) values
   ('Shikhar', 'Dhawan', 33, 'Delhi', 'India');
INSERT 0 1
postgres=#

在使用INSERT INTO语句插入记录时,如果您跳过任何列名,则将插入Record,这将在您跳过的列上留下空白。

postgres=# insert into CRICKETERS 
   (First_Name, Last_Name, Country) values('Jonathan', 'Trott', 'SouthAfrica');
INSERT 0 1

如果您传递的值的顺序与表中它们各自的列名相同,则也可以在不指定列名的情况下将记录插入表中。

postgres=# insert into CRICKETERS values('Kumara', 'Sangakkara', 41, 'Matale', 'Srilanka');
INSERT 0 1
postgres=# insert into CRICKETERS values('Virat', 'Kohli', 30, 'Delhi', 'India');
INSERT 0 1
postgres=# insert into CRICKETERS values('Rohit', 'Sharma', 32, 'Nagpur', 'India');
INSERT 0 1
postgres=#

将记录插入表后,您可以使用SELECT语句验证其内容,如下所示-

postgres=# SELECT * from CRICKETERS;
 first_name | last_name  | age | place_of_birth | country
------------+------------+-----+----------------+-------------
Shikhar     | Dhawan     | 33  | Delhi          | India
Jonathan    | Trott      |     |                | SouthAfrica
Kumara      | Sangakkara | 41  | Matale         | Srilanka
Virat       | Kohli      | 30  | Delhi          | India
Rohit       | Sharma     | 32  | Nagpur         | India
(5 rows)

使用Python插入数据

psycopg2的游标类提供了一个名为execute()方法的方法。此方法接受查询作为参数并执行它。

因此,要使用Python将数据插入PostgreSQL中的表中-

  • 导入psycopg2软件包。

  • 通过将用户名,密码,主机(可选的默认值:localhost)和数据库(可选)作为参数传递给connect()方法,创建连接对象。

  • 通过将false设置为属性autocommit的值来关闭自动提交模式。

  • psycopg2库的Connection类的cursor()方法返回一个游标对象。使用此方法创建一个游标对象。

  • 然后,通过将INSERT语句作为参数传递给execute()方法来执行INSERT语句。

以下Python程序在PostgreSQL数据库中创建一个名称为EMPLOYEE的表,并使用execute()方法在其中插入记录-

import psycopg2

#Establishing the connection
conn = psycopg2.connect(
   database="mydb", user='postgres', password='password', host='127.0.0.1', port= '5432'
)

#Setting auto commit false
conn.autocommit = True

#Creating a cursor object using the cursor() method
cursor = conn.cursor()

# Preparing SQL queries to INSERT a record into the database.
cursor.execute('''INSERT INTO EMPLOYEE(FIRST_NAME, LAST_NAME, AGE, SEX, INCOME) 
   VALUES ('Ramya', 'Rama priya', 27, 'F', 9000)''')
cursor.execute('''INSERT INTO EMPLOYEE(FIRST_NAME, LAST_NAME, AGE, SEX, INCOME) 
   VALUES ('Vinay', 'Battacharya', 20, 'M', 6000)''')
cursor.execute('''INSERT INTO EMPLOYEE(FIRST_NAME, LAST_NAME, AGE, SEX, INCOME) 
   VALUES ('Sharukh', 'Sheik', 25, 'M', 8300)''')
cursor.execute('''INSERT INTO EMPLOYEE(FIRST_NAME, LAST_NAME, AGE, SEX, INCOME) 
   VALUES ('Sarmista', 'Sharma', 26, 'F', 10000)''')
cursor.execute('''INSERT INTO EMPLOYEE(FIRST_NAME, LAST_NAME, AGE, SEX, INCOME) 
   VALUES ('Tripthi', 'Mishra', 24, 'F', 6000)''')

# Commit your changes in the database
conn.commit()

print("Records inserted........")

# Closing the connection
conn.close()

输出

Records inserted........