from sklearn.preprocessing import MultiLabelBinarizer
labellist=[['红色'],['绿色'],['红色','绿色']]
mlb=MultiLabelBinarizer()
label_num=mlb.fit(labellist)
print(f'label的具体名称为{label_num.classes_}')
train_y=[]
for label in labellist:
matrix=mlb.transform([label])
train_y.append(matrix[0])
print(train_y)
mlb.fit_transform([label])#得到的是维度不一致的数组