Векторизация массива numpy

@Krutko · Регистрация: 13.11.2015

Author24 — интернет-сервис помощи студентам

Есть массив numpy, который содержит номера классов 1,2,3 необходимо представить эти классы в следующем виде:
1=[1,0,0]
2=[0,1,0]
3=[0,0,1]
Внизу массив классов 105Х1

Python

[[3]
 [2]
 [1]
 [1]
 [3]
 [2]
 [3]
 [3]
 [1]
 [2]
 [2]
 [1]
 [2]
 [2]
 [2]
 [2]
 [2]
 [2]
 [3]
 [3]
 [1]
 [1]
 [3]
 [3]
 [2]
 [2]
 [2]
 [3]
 [3]
 [2]
 [1]
 [1]
 [1]
 [3]
 [1]
 [1]
 [2]
 [2]
 [2]
 [1]
 [1]
 [1]
 [3]
 [2]
 [3]
 [1]]

@vrm2 · 08.01.2019, 18:36

А что значит запись?

Код

1=[1,0,0]

На первом месте стоит элемент первого класса (равно 1), а на других местах другие классы (поэтому равно 0)?
Расшифруйте, что хотите в итоге получить.

@dondublon · 09.01.2019, 10:53

Странно, куда-то пропал мой ответ. Решение для произвольного случая:

Python

import numpy as np
arr1 = np.array([[1,0,0],
                 [0,1,0],
                 [0,0,1]])
 
arr2 = np.array([[2],
                 [1],
                 [0],
                 [1],
                 [2]])
 
n = arr2.shape[0]
arr21 = arr2.reshape((-1, ))
print(arr1[arr21])

Если массивы на замену такие, как у вас, т. е. у i-го массива на i-м месте стоит единица, остальные нули - то можно проще.

@vrm2 431 / 302 / 89 Регистрация: 03.12.2015 Сообщений: 738
	08.01.2019, 18:36	2
	А что значит запись? Код 1=[1,0,0] На первом месте стоит элемент первого класса (равно 1), а на других местах другие классы (поэтому равно 0)? Расшифруйте, что хотите в итоге получить. 0

	09.01.2019, 10:53