Развернуть numpy массив индексов в matix

Question

У меня есть массив numpy из N целых чисел от 0 до M включительно. Я хочу рассматривать их как индексы в матрице NxM, которая содержит 1 в каждой позиции, обозначенной массивом, и 0 всюду. Например, если дано N = 4, M = 2 У меня следующий массив

[1, 0, 2, 1] 

Я хочу, чтобы получить эту матрицу

[0 1 0] 
[1 0 0] 
[0 0 1] 
[0 1 0] 

т.е. строки 0 имеет один в столбце 1, строка 1 имеет 1 в столбце 0 и т. д.

Как сделать это преобразование в numpy?

Answer 1

Для этого требуется multi-dimensional array indexing.

a = np.array([1, 0, 2, 1]) 
z = np.zeros(12, dtype=int).reshape(4,3) 
z[np.arange(a.size), a] = 1