序类别标签转化为扩展二分类标签

import numpy as np

nClass = 5
nTheta = 4
labels = [0,1,2,3,4]

def get_binary_label_matrix():
    label_matrix = np.zeros((nClass, nTheta))
    for i, lab in enumerate(labels):
        binary_labels = np.ones(nTheta, dtype=np.int32)
        for k in range(nTheta):
            if i <= k:
                binary_labels[k] = 1
            else:
                binary_labels[k] = -1
        label_matrix[i] = binary_labels
    return label_matrix

print(get_binary_label_matrix())