以下是一个Python脚本，功能是连接MySQL数据库，执行查询语句，并将查询结果保存为CSV或Excel文件。 ### 实现代码 ```python import mysql.connector import pandas as pd def query_to_file(host, user, password, database, query, output_path, file_format="csv"): """ 连接MySQL数据库，执行查询并将结果保存为CSV或Excel文件。 参数: host (str): 数据库主机地址。 user (str): 数据库用户名。 password (str): 数据库密码。 database (str): 数据库名称。 query (str): SQL查询语句。 output_path (str): 输出文件路径。 file_format (str): 文件格式，支持"csv"和"excel"，默认为"csv"。 返回: None """ try: # 连接MySQL数据库 connection = mysql.connector.connect( host=host, user=user, password=password, database=database ) # 执行查询并获取结果 df = pd.read_sql_query(query, connection) # 根据文件格式保存结果 if file_format == "csv": df.to_csv(output_path, index=False, encoding='utf-8') print(f"查询结果已保存为CSV文件: {output_path}") elif file_format == "excel": df.to_excel(output_path, index=False, encoding='utf-8') print(f"查询结果已保存为Excel文件: {output_path}") else: print("不支持的文件格式，请选择'csv'或'excel'。") except Exception as e: print(f"发生错误: {e}") finally: # 关闭数据库连接 if 'connection' in locals() and connection.is_connected(): connection.close() # 示例用法 if __name__ == "__main__": db_config = { "host": "localhost", # 替换为你的数据库主机地址 "user": "root", # 替换为你的数据库用户名 "password": "password", # 替换为你的数据库密码 "database": "test_db" # 替换为你的数据库名称 } sql_query = "SELECT * FROM your_table;" # 替换为你的SQL查询语句 output_file = "output.csv" # 替换为你希望生成的文件名 # 调用函数生成CSV文件 query_to_file(**db_config, query=sql_query, output_path=output_file, file_format="csv") ``` --- ### 代码解释 1. **`mysql.connector.connect()`**: - 使用`mysql.connector`库连接到MySQL数据库。 - 需要提供数据库的`host`、`user`、`password`和`database`信息。 2. **`pd.read_sql_query(query, connection)`**: - 使用Pandas库的`read_sql_query`方法执行SQL查询，并将结果加载到DataFrame中。 3. **文件保存**: - 如果`file_format`为`csv`，则使用`df.to_csv()`将DataFrame保存为CSV文件。 - 如果`file_format`为`excel`，则使用`df.to_excel()`将DataFrame保存为Excel文件。 - `index=False`表示不将DataFrame的索引列写入文件。 - `encoding='utf-8'`确保输出文件使用UTF-8编码，避免中文字符或其他特殊字符出现乱码。 4. **异常处理**: - 使用`try-except-finally`结构捕获可能的异常（如数据库连接失败、查询错误等），并在最后确保关闭数据库连接。 5. **示例用法**: - 在`if __name__ == "__main__":`部分提供了一个简单的示例，演示如何调用`query_to_file`函数。 --- ### 注意事项 1. **安装依赖**: - 确保已安装`mysql-connector-python`和`pandas`库。可以通过以下命令安装： ```bash pip install mysql-connector-python pandas openpyxl ``` 2. **支持的文件格式**: - 当前脚本支持将查询结果保存为CSV或Excel文件。如果需要其他格式（如JSON、HTML等），可以扩展代码逻辑。 3. **安全性**: - 避免在代码中硬编码敏感信息（如数据库密码）。可以使用环境变量或配置文件来存储这些信息。 ---