如何根据文件名将多个CSV文件插入对应不同数据库表？

看起来你遇到的问题是典型的硬编码表名导致的——你代码里大概率把目标表名固定成了room_icslab1，没有根据CSV文件名动态生成对应的表名。下面我给你拆解解决思路和具体代码示例：

1. 核心问题定位 #

你现在的代码逻辑应该是类似这样的（伪代码）：

# 错误示例：固定死表名
table_name = "room_icslab1"
for csv_file in selected_files:
    # 读取CSV数据
    data = read_csv(csv_file)
    # 插入固定表
    insert_into_db(table_name, data)

这种写法不管你选哪个CSV，都会往同一个表里插，自然不符合需求。

2. 正确的动态表名生成逻辑 #

我们需要从CSV文件名中提取标识，拼接成对应的表名：

• 从ICSLAB1.csv提取icslab1 → 拼接为room_icslab1
• 从ICSLAB2.csv提取icslab2 → 拼接为room_icslab2

以Python为例，用os模块处理文件名：

import os

# 遍历你选中的所有CSV文件
for csv_file_path in ["ICSLAB1.csv", "ICSLAB2.csv", "ICSLAB3.csv", "ICSLAB4.csv"]:
    # 1. 获取纯文件名（去掉路径）
    file_name = os.path.basename(csv_file_path)  # 得到 "ICSLAB1.csv"
    # 2. 去掉后缀，转小写（适配数据库表名的大小写习惯）
    file_prefix = os.path.splitext(file_name)[0].lower()  # 得到 "icslab1"
    # 3. 动态生成目标表名
    target_table = f"room_{file_prefix}"  # 得到 "room_icslab1"
    
    # 接下来就是读取CSV数据并插入target_table的逻辑
    # 示例：用pandas读取CSV并插入数据库
    import pandas as pd
    df = pd.read_csv(csv_file_path)
    df.to_sql(target_table, con=your_db_connection, if_exists="append", index=False)

3. 额外的安全与验证建议 #

为了避免非法文件名导致的SQL注入或错误表名，最好加一层验证：

# 定义允许的表名列表，确保只有预期的表能被操作
allowed_tables = {"room_icslab1", "room_icslab2", "room_icslab3", "room_icslab4"}

if target_table not in allowed_tables:
    print(f"警告：非法文件名 {file_name}，跳过处理")
    continue

4. 常见坑点排查 #

• 大小写问题：如果你的数据库表名是小写的，记得把文件名前缀转小写（比如ICSLAB1→icslab1）
• 特殊文件名处理：如果文件名包含多个.（比如ICSLAB1.v2.csv），用os.path.splitext比直接split('.')[0]更稳妥
• 路径处理：如果用户选择的是带路径的文件（比如./data/ICSLAB1.csv），一定要用os.path.basename提取纯文件名，不然会把路径也带进表名里

内容的提问来源于stack exchange，提问作者Lanz Joshua