用pycharm自定义一个函数,这个函数可以向mysql中的表中插入数据
时间: 2023-06-16 17:05:30 浏览: 92
在python中使用pymysql往mysql数据库中插入(insert)数据实例
可以使用Python的MySQL库`pymysql`来连接MySQL数据库。以下是一个示例函数,可以将数据插入到MySQL表中:
```python
import pymysql
def insert_data_to_db(db_host, db_port, db_user, db_password, db_name, table_name, data):
"""
将数据插入到MySQL表中
:param db_host: 数据库主机名
:param db_port: 数据库端口号
:param db_user: 数据库用户名
:param db_password: 数据库密码
:param db_name: 数据库名称
:param table_name: 表名称
:param data: 要插入的数据,类型为tuple或list
:return: None
"""
# 连接数据库
conn = pymysql.connect(host=db_host, port=db_port, user=db_user, password=db_password, database=db_name)
cursor = conn.cursor()
# 构建SQL语句
placeholders = ', '.join(['%s'] * len(data))
columns = ', '.join(data.keys())
sql = "INSERT INTO %s (%s) VALUES (%s)" % (table_name, columns, placeholders)
# 执行SQL语句
try:
cursor.execute(sql, list(data.values()))
conn.commit()
except:
conn.rollback()
# 关闭连接
cursor.close()
conn.close()
```
调用示例:
```python
data = {'name': '张三', 'age': 18, 'gender': '男'}
insert_data_to_db('localhost', 3306, 'root', '123456', 'test', 'student', data)
```
在这个示例中,我们将数据插入到名为“student”的表中,表有三个字段:`name`、`age`、`gender`。数据以字典形式传递给函数。
阅读全文