def recursive_function(config, save_filePath, now_day, save_dict): for tj_config in config.tj_config.keys(): tj_config_data = self.pickUpType(self.getData(),tj_config) #按type提取数据 save_typefilename = tj_config +"_"+ now_day + ".csv" save_typefile = os.path.join(save_filePath,save_typefilename) #该类的保存文件名 TianJiIocs.save_Csv(tj_config_data, save_typefile) # 生成Type文件 #保存数据到字典 save_dict[tj_config] = {} # json save_dict[tj_config]["dataFile"] = save_typefile save_dict[tj_config]["count"] = f'{tj_config}类-数据总量:{tj_config_data.shape[0]}'

    #威胁类
    category_config = config.tj_config[tj_config]["category"]

    for category in category_config.keys():
        category_data = self.pickUpCategory(tj_config_data,category)
        save_categoryFilename = tj_config + "_" + category + "_" + now_day + ".csv"
        save_categoryFile = os.path.join(save_filePath, save_categoryFilename)  # 该威胁类型的保存文件名
        TianJiIocs.save_Csv(category_data, save_categoryFile)  # 生成威胁类文件
        #调用递归函数
        recursive_function(config, save_filePath, now_day, save_dict)
将代码变成递归 for tj_config in configtj_configkeys tj_config_data = selfpickUpTypeselfgetDatatj_config #按type提取数据 save_typefilename = tj_config +_+ now_day + csv

原文地址: http://www.cveoy.top/t/topic/i9CL 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录