У меня есть матрица np, и я хочу преобразовать ее в трехмерный массив с одним горячим кодированием элементов в качестве третьего измерения. Есть ли способ обойтись без зацикливания на каждой строке, например

a=[[1,3],
   [2,4]]

должно быть сделано в

b=[[1,0,0,0], [0,0,1,0],
   [0,1,0,0], [0,0,0,1]]