import pandas as pd

def list_topic(total_num, str1):

    """

    生成多个主题

    :param total_num: 总的主题数量

    :param str1: 主题名称中的固定字母，如A1,A2,A3,A4....

    :return: 返回主题列表

    """

    list_topic = []

    num = 1

    for j in range(total_num):

        topic_name = str1 + str(num)

        list_topic.append(topic_name)

        num += 1

    return list_topic

def dir_name(dir_num, str1):

    """

    生成多个主题

    :param total_num: 总的主题数量

    :param str1: 目录名称中的固定字母，如A1,A2,A3,A4....

    :return: 返回目录列表

    """

    list_dir = []

    num = 1

    for j in range(dir_num):

        topic_name = str1 + str(num)

        list_dir.append(topic_name)

        if j != 0:

            # 每隔100，对目录加1，比如第1个100，目录名都为A1,第2个100，目录名都为A2

            if j % 100 == 0:

                num += 1

    return list_dir

# print(dir_name(200, "A"))

if __name__ == "__main__":

    all_dir = dir_name(1000, "目录")

    all_topic = list_topic(1000, "topic")

    data = {"目录": all_dir, "主题": all_topic}

    df = pd.DataFrame(data)

    df.to_excel("ECHAT TOPIC 3.xlsx", sheet_name="对话主题", index=False, header=True)

import pandas as pd

def dir_name(dir_num, str1):

    """

    生成多个具有层级关系的目录

    :param total_num: 总的目录数量

    :param str1: 目录名称中的固定字母，如A1,A2,A3,A4....

    :return: 返回目录列表

    """

    list_dir = []

    num = 1

    topic_name = str1 + str(num)

    for j in range(1, dir_num):

            topic_name = str1 + str(num)

            list_dir.append(topic_name)

            if j % 100 == 0:

                num += 1

                # 每达到100，目录名自增1，并且把当前目录和列表中最后一个目录拼接起来，如目录1>目录1

                topic_name = ">".join([list_dir[-1], topic_name])

                for sub_dir in range(100):

                    # 每达到100后，之后的100个目录都是新的具有层级关系的目录

                    list_dir.append(topic_name)

    return list_dir

print(dir_name(401, "目录"))

print(len(dir_name(401, "目录")))

# str_list = ['Hello','wordl']

# str_join1 = ''.join(str_list)

# str_join2 = '>'.join(str_list)

# print(str_join1)

# print(str_join2)