'''
Description: 
Author: CrayonXiaoxin
Date: 2024-09-10 09:49:17
LastEditTime: 2024-09-10 15:50:30
LastEditors:  
'''
import os
import openpyxl

if __name__ == '__main__':
    path = os.path.dirname(__file__)
    os.chdir(path)
    print(path)
    # 检查文件是否存在
    file_name = input("请输入文件名：")
    if not os.path.exists(file_name):
        raise FileNotFoundError(f"文件 {file_name} 不存在！")

    workbook = openpyxl.load_workbook(file_name)
    print(workbook.sheetnames)
    sheet1 = input("请输入主表名字：")
    # 获取 Sheet1（假设其名字是 '1'）
    if sheet1 not in workbook.sheetnames:
        raise ValueError("找不到名为 %s 的工作表"%sheet1)

    sheet = workbook[sheet1]
    rows = sheet.max_row
    column = sheet.max_column

    # 获取表头
    topName = []
    for i in range(1, column+1):
        topName.append(sheet.cell(1, i).value)
    names = []
    # 自定义分类名称
    for i in range(1,1000):
        name = input("请输入要整理的名字：(如果想结束请输入#)")
        if name == '#':
            break
        names.append(name)
    

    # 创建新的 Sheet
    for name in names:
        if name not in workbook.sheetnames:
            workbook.create_sheet(name)

    # 按分类写入数据
    for i in range(2, rows+1):  # 从第2行开始读取，因为第1行是表头
        name = sheet.cell(i, 2).value
        if name in names:
            viceSheet = workbook[name]
            values = [sheet.cell(i, j).value for j in range(1, column+1)]
            vicerows = viceSheet.max_row
            if vicerows == 0:
                # 如果新建的表是空的，首先写入表头
                for l in range(1, column+1):
                    viceSheet.cell(1, l, topName[l-1])
                vicerows = 1
            for k in range(1, column+1):
                viceSheet.cell(vicerows + 1, k, values[k-1])

    # 保存并关闭文件
    workbook.save(file_name)
    workbook.close()

a = input('输入')//input()返回的数据类型是str，需要进行转换
a = int(a)
print(a)

if a < 10:
	print(a)
elif a > 10 and a < 20:
	print(a)
else
	print(a)

//简单匹配
score = 'B'
match score:
    case 'A':
        print('A')
    case 'B':
        print('B')
//复杂匹配
age = 15
match age:
    case x if x < 10://当age < 10 成立时匹配，且赋值给x
        print(1)
    case 10:
        print(10)
    case 11 | 12 | 13://匹配多个值
    	print(11~13)
        
 
//循环
names = ['a','b']
for name in names:
    print(name)

sum = 0
for x in range(100):
    sum = sum + x
print(sum)

while n > 0:
    sum = sum + n
    n = n - 1
print(sum)

classmate = ['a','b','c']
len = len(classmate)//获取元素个数
s = classmate[0] //索引访问，下标从0开始
s = classmate[-1] //获取最后一个元素 
s = classmate[-2]//倒数第二个
classmate.append('d')//追加元素到末尾
classmate.insert(1,'g')//往索引为1的地方插入元素
classmate.pop()//删除末尾元素
classmate.pop(i)//删除索引为i的元素
classmate[1] = 'f'//修改元素

t = (1,2)
t = (1,)// ！只有一个元素是需要加一个逗号消除歧义