Python 创建数据库并插入数据
时间: 2023-11-05 16:08:14 浏览: 116
首先,需要安装并导入 Python 的数据库操作模块。常用的有 MySQLdb、pymysql、sqlite3 等。
以 sqlite3 为例,首先需要创建一个数据库文件,并连接到该数据库:
```python
import sqlite3
# 创建或连接到数据库
conn = sqlite3.connect('example.db')
```
接着,可以创建表格并插入数据:
```python
# 创建表格
conn.execute('''CREATE TABLE IF NOT EXISTS users
(id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
age INTEGER NOT NULL);''')
# 插入数据
conn.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))
conn.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Bob', 28))
# 提交更改并关闭连接
conn.commit()
conn.close()
```
以上代码中,首先使用 `CREATE TABLE` 语句创建了 `users` 表格,其中包含 `id`、`name` 和 `age` 三个字段。`id` 设为主键,并自增长;`name` 和 `age` 分别为文本和整数类型,且不能为空。
然后使用 `INSERT INTO` 语句向表格中插入数据。注意,插入时必须按照表格中字段的顺序传入数据,用问号 `?` 占位符表示。最后,使用 `commit()` 方法提交更改,并使用 `close()` 方法关闭连接。
如果需要查询数据,可以使用 `SELECT` 语句。例如:
```python
# 查询数据
cursor = conn.execute("SELECT id, name, age FROM users")
for row in cursor:
print("id = {}, name = {}, age = {}".format(row[0], row[1], row[2]))
```
以上代码中,使用 `execute()` 方法执行 `SELECT` 语句,并使用 `for` 循环遍历查询结果。每一行数据都是一个元组,可以通过下标访问其中的字段值。
阅读全文