sqlalchemy ORM進階－ 批量插入數據

 

參考：

https://www.jb51.net/article/49789.htm

https://blog.csdn.net/littlely_ll/article/details/82706874

 

import pandas as pd
import sqlalchemy

df = pd.DataFrame()
print(df)

結果：

Empty DataFrame
Columns: []
Index: []

 

插入1條數據

 

import sqlalchemy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,String,Integer,Float,TIMESTAMP

from demo.demogetConnection import GetConnection

session = GetConnection.session
# engine =GetConnection.engine

Base = declarative_base()
class User(Base):
    __tablename__="user"
    id = Column(Integer,primary_key=True)
    name = Column(String(45))

    def __repr__(self):
        return "<User(id='%s',name='%s')>"%(self.id,self.name)

#插入單條數據：
new_user = User(name='lily')
#添加數據，但還沒有提交，出錯還可以使用rollback撤回操作
session.add(new_user)
#提交到數據，這一步才是真正的將數據插入到數據庫中了
session.commit()

結果

 

利用pandas批量插入數據

方式一：使用pandas_obj.to_sql()

import pandas as pd
import sqlalchemy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,String,Integer,Float,TIMESTAMP


from demo.demogetConnection import GetConnection

session = GetConnection.session
engine =GetConnection.engine

Base = declarative_base()
class User(Base):
    __tablename__="user"
    id = Column(Integer,primary_key=True)
    name = Column(String(45))

    def __repr__(self):
        return "<User(id='%s',name='%s')>"%(self.id,self.name)



df = pd.DataFrame({"name":["Jon","Mary","Tom"]})
df.to_sql("user",con=engine,if_exists="append",index=False)

 

 在使用to_sql時注意if_exists參數，如果是replace的話它會先drop掉表，然后再創建表，最后插入數據

結果：

 

第2種方法，使用for循環

import sqlalchemy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,String,Integer,Float,TIMESTAMP

from demo.demogetConnection import GetConnection

session = GetConnection.session
engine =GetConnection.engine

Base = declarative_base()
class User(Base):
    __tablename__="user"
    id = Column(Integer,primary_key=True)
    name = Column(String(45))

    def __repr__(self):
        return "<User(id='%s',name='%s')>"%(self.id,self.name)


#原生方法，批量插入
a=[]
for i in range(3):
    a.append({"name":"曹操%s"%i})

print(a)
session.execute(User.__table__.insert(),a)
session.commit()

打印的a的結果：

[{'name': '曹操0'}, {'name': '曹操1'}, {'name': '曹操2'}]

查詢表的結果，已經插入成功了：