在数据库中建立新表并导入数据 通过 PyMySQL 库存储实训 2 提取的网页内容,在 MySQL 的 test 库中建立一个新表, 并将提取的文本内容存入该表内,之后查询该表内容,确认是否存储成功。
时间: 2024-10-24 11:05:47 浏览: 14
在Python中使用PyMySQL库操作MySQL数据库,可以按照以下步骤来创建新表并导入数据:
1. 导入库:
```python
import pymysql
```
2. 连接到数据库(假设数据库地址、用户名、密码和数据库名称已知):
```python
db = pymysql.connect(host="数据库地址", user="用户名", password="密码", database="test")
cursor = db.cursor()
```
3. 创建新表,例如,如果你的字段结构类似文章标题和内容:
```python
create_table_query = """
CREATE TABLE IF NOT EXISTS article (
id INT AUTO_INCREMENT PRIMARY KEY,
title VARCHAR(255) NOT NULL,
content TEXT
);
"""
cursor.execute(create_table_query)
db.commit() # 确保更改生效
```
4. 将提取的数据插入新表:
```python
# 假设 `data` 是你已经提取的文本内容
for i, row in enumerate(data):
insert_data_query = f"INSERT INTO article (title, content) VALUES ('{row[0]}', '{row[1]}')"
cursor.execute(insert_data_query)
if i % 100 == 0: # 批量插入以提高效率
db.commit()
db.commit() # 插入完最后一行后提交
```
5. 查询新表内容来确认数据是否已成功存储:
```python
query = "SELECT * FROM article;"
cursor.execute(query)
rows = cursor.fetchall()
for row in rows:
print(f"ID: {row[0]}, Title: {row[1]}, Content: {row[2]}")
# 关闭连接
cursor.close()
db.close()
```
阅读全文