Как мне векторизовать это выражение с помощью Numpy? ⇐ Python

[Anonymous]

Я пытаюсь вычислить «матрицу» массива ниже. Как я могу это сделать, используя векторизованные функции NumPy?

Код: Выделить всё

x = np.array([
[2, 1, 0],
[1, 1, 0],
[3, 2, 1],
[1, 0, 0],
[2, 3, 0]
])

y = np.array([
[405, 200, 150],
[200, 300, 150],
[150, 100, 105],
[425, 200, 250],
[500, 620, 300]
])

matrix = np.zeros((5,3,4))
for i in range(5):
for j in range(3):
matrix[i,j,x[i,j]] = y[i,j]

Я пробовал это:

Код: Выделить всё

vmatrix = np.zeros((5,3,4))

vmatrix[:,:,x] = y

Но это не работает...

Подробнее здесь: https://stackoverflow.com/questions/790 ... with-numpy