产生多个目录-每个目录下多个对话主题

Posted 2021-03-08 come202011

产生多个目录，每个目录下产生多个对话主题，写入到表格中

 

# -*- coding: utf-8 -*-
# @Time : 2020/2/10 14:18
# @File : write_excel2.py
# @Author: Hero Liu
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)

 

 

产生具有层级关系的多级目录

# -*- coding: utf-8 -*-
# @Time : 2020/2/10 14:18
# @File : write_excel2.py
# @Author: Hero Liu
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)