### 3.1 插入数据到 MySQL 数据库 在 Python 中，可以通过 `mysql-connector-python` 或 `pymysql` 等库连接 MySQL 数据库，并执行 SQL 插入语句。以下是一个插入数据到 MySQL 的示例： ```python import mysql.connector # 建立数据库连接 cnx = mysql.connector.connect( host="localhost", user="root", password="password", database="test_db" ) cursor = cnx.cursor() # 定义插入语句 insert_query = """ INSERT INTO users (name, age, email) VALUES (%s, %s, %s) """ # 插入数据 data_to_insert = [ ("Alice", 25, "alice@example.com"), ("Bob", 30, "bob@example.com"), ("Charlie", 22, "charlie@example.com") ] cursor.executemany(insert_query, data_to_insert) cnx.commit() cursor.close() cnx.close() ``` 该代码使用 `executemany` 方法实现批量插入，适用于一次性插入多条记录的场景。 ### 3.2 将数据保存到 YAML 文件 YAML 是一种结构化配置文件格式，适合存储嵌套数据结构。在 Python 中，可以使用 `PyYAML` 库进行 YAML 文件的读写操作。以下是将数据写入 YAML 文件的示例： ```python import yaml # 定义要保存的数据 data = { "users": [ {"name": "Alice", "age": 25, "email": "alice@example.com"}, {"name": "Bob", "age": 30, "email": "bob@example.com"}, {"name": "Charlie", "age": 22, "email": "charlie@example.com"} ] } # 写入 YAML 文件 with open("users.yaml", "w", encoding="utf-8") as file: yaml.dump(data, file, allow_unicode=True) ``` 该代码将用户信息存储到 `users.yaml` 文件中，结构清晰，便于后续读取和维护。 ### 3.3 读取 YAML 文件中的数据 在后续操作中，可以从 YAML 文件中读取存储的数据。以下是一个读取 YAML 文件的示例： ```python import yaml # 读取 YAML 文件 with open("users.yaml", "r", encoding="utf-8") as file: data = yaml.safe_load(file) # 输出读取到的数据 print(data) ``` `yaml.safe_load()` 方法用于安全地加载 YAML 文件内容，避免执行任意代码的风险。此方法适用于大多数配置文件的读取场景。 ### 3.4 封装 YAML 文件读取功能 为了提高代码的复用性，可以将 YAML 文件的读取功能封装为一个类。以下是一个封装示例： ```python import yaml import os class YamlLoader: def __init__(self, file_path): if os.path.exists(file_path): self.file_path = file_path else: raise FileNotFoundError(f"未找到文件路径: {file_path}") self.data = self.read_yaml() def read_yaml(self): with open(self.file_path, 'r', encoding='utf-8') as f: return yaml.safe_load(f) def get_data(self, key=None): if key is None: return self.data else: return self.data.get(key) # 使用示例 if __name__ == "__main__": loader = YamlLoader("users.yaml") yaml_data = loader.get_data() print(yaml_data) ``` 该类支持读取整个 YAML 文件内容，也支持按指定键提取数据，提高了灵活性和可维护性。 ###