python操作postgresql数据库

用python将dataframe直接插入postgresql数据库

1、连接postgresql数据库

import psycopg2
 
#连接数据库
conn = psycopg2.connect(database = '数据库名字', user = '用户名', password = '密码', host = '主机名', port = '端口号')
 
curs=conn.cursor()

2、查看数据库中某一个表的数据（以dataframe的形式查看）

#编写Sql（获取列名）
col = "SELECT string_agg(column_name, ',') FROM information_schema.columns where table_schema='public' and table_name='表名字'"
 
#数据库中执行sql命令
curs.execute(col)
 
#获得数据
cols = curs.fetchall()

#编写Sql
sql = 'select * from 表名字'
 
#数据库中执行sql命令
curs.execute(sql)
 
#获得数据
data = curs.fetchall()

import pandas as pd
# 将查出的数据转换为dataFrame
df = pd.DataFrame(data).reset_index(drop=True)

# 为表添加columns列名
columns = list(cols[0])[0].split(',')
df.columns = columns
df.head(10)

3、将需要插入的dataframe插入postgresql数据库

import psycopg2.extras
if len(dataframe的名字) > 0:

    df_columns = list(dataframe的名字)
    columns = ",".join(df_columns)
    values = "VALUES({})".format(",".join(["%s" for _ in df_columns])) 
    insert_stmt = "INSERT INTO {} ({}) {}".format('所需要插入的表的名字',columns,values)
    curs = conn.cursor()
    psycopg2.extras.execute_batch(curs, insert_stmt, dataframe的名字.values)
    conn.commit()
    curs.close()

那些更新操作啊，删除操作啊，改一下sql语句就行，这里就不多做介绍了。简单介绍两个例子好了。

4、更新数据

#编写Sql(重命名列名)
rename = "alter table 表名字 rename column 原列名 to 新列名" 
 
#数据库中执行sql命令
curs.execute(rename)

#提交数据    
conn.commit()

5、删除数据

#编写Sql（删除id列为1的相关数据）
#delete = "delete from 表名字 where id=1" 
 
#数据库中执行sql命令
#curs.execute(delete)

#提交数据    
#conn.commit()