python 用 pymysql 向数据表插入数据

一、 加载库、封装函数

import pandas as pd import pymysql def mycursor(db_name=None): '''连接数据库，创建游标''' config = dict(zip(['host', 'user', 'port', 'password'], ['192.168.137.155', 'shanger', 3306, '0123'])) config.update(database=db_name) connection = pymysql.connect(**config) cursor = connection.cursor() return cursor def use(db_name): '''切换数据库，返回游标'''
    return mycursor(db_name) def insert_many(table, data): '''向全部字段插入数据''' val = '%s, ' * (len(data[0])-1) + '%s' sql = f'insert into {table} values ({val})' cursor.executemany(sql, data) cursor.connection.commit() def query(sql): '''以数据框形式返回查询据结果''' cursor.execute(sql) data = cursor.fetchall()  # 以元组形式返回查询数据
    header = [t[0] for t in cursor.description] df = pd.DataFrame(list(data), columns=header)  # pd.DataFrem 对列表具有更好的兼容性
    return df
 def select_database(): '''查看当前数据库''' sql = 'select database();'
    return query(sql) def show_tables(): '''查看当前数据库中所有的表''' sql = 'show tables;'
    return query(sql) def select_all_from(table): sql = f'select * from {table};'
    return query(sql)

 

二、插入数据

1、选则数据库、查看数据

 

 2、插入数据

data = [ ('德岛', '100'), ('香川', '200'), ('爱媛', '150'), ('高知', '200'), ('福冈', '300'), ('佐贺', '100'), ('长崎', '200'), ('东京', '400'), ('群马', '50')] insert_many('poptbl', data) select_all_from('poptbl')