python根据指定列名拆分表格、或指定行数拆分表格

import pandas as pd
import tkinter as tk
from tkinter import messagebox

root = tk.Tk()
root.withdraw()  # 隐藏主窗口
root.wm_attributes('-topmost',1)  #窗口置顶

path = 'result_to_split'

# try:
df = pd.read_excel(path+'.xlsx')
y = int(input('请选择输入数字：1.根据列名拆分；2.根据固定条数拆分：'))
if y == 1:
    x = input("请输入你需要拆分的列名：")
    citys = set(df[x].unique())
    for i in citys:
        df1 = df[df[x] == i]
        df1.to_excel(path + "_" + i + '.xlsx', index=None)
    print(f"已成功拆分成{len(citys)}个表！")
    messagebox.showinfo("拆分结果", f"共{len(df)}条数据，已成功拆分成{len(citys)}个表")
else:
    num_splits = int(input("请输入每个表需要几行："))
    # df_split = pd.DataFrame()
    start_index = 0
    # 可以拆分成多少个表
    num = int(len(df) / num_splits +1) if len(df) % num_splits >0 else int(len(df) / num_splits)
    for i in range(0,num):
        df_split = df.iloc[start_index:start_index+num_splits]
        df_split.to_excel(path + "_" + str(i) + '.xlsx', index=None)
        start_index += num_splits
    print(f"共{len(df)}条数据，已成功拆分成{num}个表，每个表{num_splits}行")
    messagebox.showinfo("拆分结果", f"共{len(df)}条数据，已成功拆分成{num}个表，每个表{num_splits}行")

请选择输入数字：1.根据列名拆分；2.根据固定条数拆分：2
请输入每个表需要几行：4
共21条数据，已成功拆分成6个表，每个表4行