用pycharm自定义一个可以向使用mysql- connector模块创建的表中插入数据的函数
时间: 2023-06-16 12:05:30 浏览: 91
好的,以下是使用 `mysql-connector` 模块创建表并插入数据的示例代码:
```python
import mysql.connector
def insert_data_to_table(name, age):
# 连接 MySQL 数据库
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
# 创建数据表
mycursor = mydb.cursor()
mycursor.execute("CREATE TABLE IF NOT EXISTS customers (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), age INT)")
# 插入数据到数据表
sql = "INSERT INTO customers (name, age) VALUES (%s, %s)"
val = (name, age)
mycursor.execute(sql, val)
# 提交更改
mydb.commit()
# 输出插入成功的记录数
print(mycursor.rowcount, "记录插入成功。")
# 关闭数据库连接
mydb.close()
# 测试插入数据
insert_data_to_table("张三", 25)
```
以上代码中,`insert_data_to_table` 函数可以将传入的 `name` 和 `age` 插入到名为 `customers` 的数据表中。如果数据表不存在,则会自动创建。这个函数使用 `mysql.connector` 模块连接 MySQL 数据库,并使用 `execute` 方法执行 SQL 语句。
注意,在使用这个函数前,你需要先安装 `mysql-connector` 模块,可以使用以下命令在 PyCharm 中安装:
```bash
pip install mysql-connector-python
```
同时,你需要根据实际情况修改函数中的数据库配置,包括 `host`、`user`、`password` 和 `database`。
阅读全文