Pythonの関数作成に関する問題です。

Question

category 表現から one-hot 表現を得る関数を作成してください。
またその逆関数も作成して下さい。
category表現とone-hot表現は例えば以下のような関係です。
category [0, 1, 2, 1, 0, 2, 2, 1, 0]
one-hot
[[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.],
[ 0., 1., 0.],
[ 1., 0., 0.],
[ 0., 0., 1.],
[ 0., 0., 1.],
[ 0., 1., 0.],
[ 1., 0., 0.]]
今、category表現はintegerとして、 one-hotのスロット数はcategory表現の最大値+1とします。(例ではcategoryの最大値が２でone-hotは３スロット)

Answer 1

こんな感じでしょうか。

def one_hot(lst):
    data = []
    for n in category:
        data.append([int(n==i) for i in range(max(lst) + 1)])
    return data

def reverse_one_hot(lst):
    return [l.index(1) for l in lst]

ワンライナーだとこんな感じ

def one_hot(lst):
    return [[int(n==i) for i in range(max(lst) + 1)] for n in category]

テスト

category = [0,1,2,1,0,2,2,1,0]
oh = one_hot(category)
print(oh)
# => [[1, 0, 0], [0, 1, 0], [0, 0, 1], [0, 1, 0], [1, 0, 0], [0, 0, 1], [0, 0, 1], [0, 1, 0], [1, 0, 0]]
rev_category = reverse_one_hot(oh)
print(rev_category)
# => [0,1,2,1,0,2,2,1,0]