def set_label(category):
dic = []
# dic = dict(zip([category],range(len(category))))
for i in range(len(category)):
category[i].append(i)
# dic = dict(zip([category[i]],[i]))
# dic.update(category[i] = i)
# dic[category[i]] = i
return category
创建一个拥有两个值的列表,我的理解是给数据加标签,这个代码最终输出的是列表,所以如果要做标签,则需要将列表转为字典类型
[ [‘adult+female+singing’, 0], [‘adult+female+speaking’, 1]]
这是列表输出结果
{‘adult+female+singing’: 0, ‘adult+female+speaking’: 1}
这是字典输出结果
只需要加上
dic = dict(category)
这个代码就可以了
具体参考:Python创建字典的八种方式