题解 | 实现One-Hot编码

import numpy as np
from sklearn.preprocessing import OneHotEncoder

def to_categorical(x, n_col=None):
    if not n_col:
        n_col = np.max(x) +1
    one_hot = np.zeros(shape=(len(x), n_col))
    for i in range(len(x)):
        one_hot[i, x[i]] = 1 # 这一块还是想了一小会哈哈哈哈，我真笨

    return one_hot
    

if __name__ == "__main__":
    x = np.array(eval(input()))
    print(to_categorical(x))