(上模型)标签库拆分多个词条后导入

发布时间 2024-01-04 19:13:05作者: 不上火星不改名
import pandas as pd
import os


def split_excel(original_file, target_folder):
    # 读取原始Excel文件
    df = pd.read_excel(original_file)

    # 计算要拆分成多少个新表格
    number_of_splits = df.shape[1] // 2

    # 对每一对列进行操作
    for i in range(number_of_splits):
        # 获取当前对列
        col_start = 2 * i
        col_end = 2 * i + 1
        data_to_save = df.iloc[:, col_start:col_end + 1]

        # 设置第一行的内容
        data_to_save.columns = ["Tag名称", "英文(对应代码)"]

        # 读取第一列的第一个单元格的值(现在是第二行)作为新文件的名称
        new_file_name = str(data_to_save.iloc[1, 0]) + ".xlsx"
        new_file_path = os.path.join(target_folder, new_file_name)

        # 保存到目标文件夹
        data_to_save.to_excel(new_file_path, index=False)


# 用户输入原始文件和目标文件夹
original_file = input("请输入原表格的完整路径:")
target_folder = input("请输入目标表格存放文件夹的路径:")

# 确保目标文件夹存在
if not os.path.exists(target_folder):
    os.makedirs(target_folder)

split_excel(original_file, target_folder)