将实验数据存储到 MySQL 数据库 - Python 代码实现

{"title": "将实验数据存储到 MySQL 数据库 - Python 代码实现", "description": "本程序实现将三种格式的实验数据文件（txt、csv、excel）存储到 MySQL 数据库中，包括流程图、函数设计、代码实现和完整程序示例。", "keywords": "实验数据存储, MySQL, Python, txt, csv, excel, 数据库操作, 数据处理", "content": ""流程图如下：\n\n\\n开始 -> 选择文件类型 -> 输入数据库信息 -> 调用相应函数 -> 存储数据 -> 结束\\n\n\n函数接口设计如下：\n\n1. txt_to_mysql 函数\n\n 函数名：txt_to_mysql\n 作用：将 txt 文件中的数据存储到 MySQL 数据库中\n 输入：txt 文件路径，数据库连接信息\n 输出：无\n\n2. csv_to_mysql 函数\n\n 函数名：csv_to_mysql\n 作用：将 csv 文件中的数据存储到 MySQL 数据库中\n 输入：csv 文件路径，数据库连接信息\n 输出：无\n\n3. excel_to_mysql 函数\n\n 函数名：excel_to_mysql\n 作用：将 excel 文件中的数据存储到 MySQL 数据库中\n 输入：excel 文件路径，数据库连接信息\n 输出：无\n\n编码实现如下：\n\npython\\nimport pymysql\\nimport pandas as pd\\n\n\ndef txt_to_mysql(txt_file, host, port, user, password, database, table):\\n # 读取 txt 文件内容\\n with open(txt_file, 'r') as file: \\n data = file.read()\\n \\n # 连接数据库\\n conn = pymysql.connect(host=host, port=port, user=user, password=password, database=database)\\n cursor = conn.cursor()\\n \\n # 创建表\\n cursor.execute(f"CREATE TABLE IF NOT EXISTS {table} (content TEXT)")\\n \\n # 插入数据\\n cursor.execute(f"INSERT INTO {table} (content) VALUES ('{data}')")\\n \\n # 提交并关闭连接\\n conn.commit()\\n cursor.close()\\n conn.close()\\n\n\ndef csv_to_mysql(csv_file, host, port, user, password, database, table):\\n # 读取 csv 文件内容\\n data = pd.read_csv(csv_file)\\n \\n # 连接数据库\\n conn = pymysql.connect(host=host, port=port, user=user, password=password, database=database)\\n cursor = conn.cursor()\\n \\n # 创建表\\n columns = ','.join(data.columns)\\n cursor.execute(f"CREATE TABLE IF NOT EXISTS {table} ({columns})")\\n \\n # 插入数据\\n for _, row in data.iterrows():\\n values = ','.join([f"'{value}'"] for value in row)\\n cursor.execute(f"INSERT INTO {table} ({columns}) VALUES ({values})")\\n \\n # 提交并关闭连接\\n conn.commit()\\n cursor.close()\\n conn.close()\\n\n\ndef excel_to_mysql(excel_file, host, port, user, password, database, table):\\n # 读取 excel 文件内容\\n data = pd.read_excel(excel_file)\\n \\n # 连接数据库\\n conn = pymysql.connect(host=host, port=port, user=user, password=password, database=database)\\n cursor = conn.cursor()\\n \\n # 创建表\\n columns = ','.join(data.columns)\\n cursor.execute(f"CREATE TABLE IF NOT EXISTS {table} ({columns})")\\n \\n # 插入数据\\n for _, row in data.iterrows():\\n values = ','.join([f"'{value}'"] for value in row)\\n cursor.execute(f"INSERT INTO {table} ({columns}) VALUES ({values})")\\n \\n # 提交并关闭连接\\n conn.commit()\\n cursor.close()\\n conn.close()\\n\n\ndef main():\\n file_type = input("请选择文件类型（1-txt，2-csv，3-excel）：")\\n txt_file = "suesnews_result.txt"\\n csv_file = "当当搜索关键词_单页面.csv"\\n excel_file = "当当搜索关键词_多页面.csv"\\n host = input("请输入数据库主机名：")\\n port = input("请输入数据库端口号：")\\n user = input("请输入数据库用户名：")\\n password = input("请输入数据库密码：")\\n database = input("请输入数据库名：")\\n table = input("请输入数据表名：")\\n \\n if file_type == "1":\\n txt_to_mysql(txt_file, host, port, user, password, database, table)\\n elif file_type == "2":\\n csv_to_mysql(csv_file, host, port, user, password, database, table)\\n elif file_type == "3":\\n excel_to_mysql(excel_file, host, port, user, password, database, table)\\n else: \\n print("文件类型选择错误！")\\n \\n\n\nif __name__ == "__main__":\\n main()\\n\n\n完整的程序可以根据实际情况进行调整和优化。