有如下list:
predict_labels = [0 1 0 1 0 0 0 0 0 0 1 1 0 0 0 0 0 1 0 1 0 1]
此时需要取出所有0和1对应的索引存入一个字典,看操作:
speaker_dict = dict()
for i in range(speaker_number):
speaker_voice = []
speaker_dict[i] = [j for j, label in enumerate(predict_labels) if label == i]
此处speaker_number代表说话人数量,0和1则2,如上操作即可,相比循环list,性能好一点。
如果只需取出0的第一个索引,则可以:
index = predict_labels.index(0)