general one-hot dictionary and encoding

one_hot_encoding.py

import numpy as np

def onehot_dict(input):
    onehotd = {k:v for v,k in enumerate(set(input))}
    return onehotd

def onehot_encoder(input,onehotd):
    dim = len(set(input))
    n = len(input)
    
    output = np.zeros((n,dim))
    for ind, i in enumerate(input):
        output[ind][onehotd[i]]=1
    return output