37,719
社区成员
发帖
与我相关
我的任务
分享
def seprateByClass(dataset):
seprate_dict = {}
info_dict = {}
for vector in dataset:
if vector[-1] not in seprate_dict:
seprate_dict[vector[-1]] = []
info_dict[vector[-1]] = 0
seprate_dict[vector[-1]].append(vector)
info_dict[vector[-1]] +=1
return seprate_dict,info_dict
sd,id=seprateByClass([[0.1,0.6,'label1'],[0.6,0.9,'label1'],[200,855,'label2']])
print("seprate_dict:{}\ninfo_dict:{}".format(sd,id))