numpy создать новый 2D-массив из существующего массива, значения столбца в виде индексов

Question

Итак, у меня есть массив с множеством чисел (4950,3). Из этого массива мне нужно создать новый массив (матрицы), значения и индексы которого взяты из существующий массив.

В частности,

Array1 = массив ([[0, 1, 0,4284256], [0, 2., 0,38090856], [1., 2., 0,32901526]]) ,

new Array2 [0,1] = 0.4284256 , Array2[0,2]=0.38090856 and Array2[1,2]= 0.32901526 

Сейчас я использую для цикла, но я имею в виду, там должен быть эффективным способом. A1 and A2 values

for i in range(A1.shape[0]): 
    A2[ A1[i,0], A1[i,1] ] = A1[i,2] 

Answer 1

Это то, что вы ищете? Вы можете напрямую индексировать значения одного массива, используя значения Integer из другого.

import numpy as np 
Array1 = np.array([ [ 0.,1.,0.4284256 ], [0.,2.,0.38090856], [1., 2.,0.32901526] ]) 
Array2 = np.zeros(Array1.shape) 
Array2[np.int_(Array1[:,0]),np.int_(Array1[:,1])] = Array1[:,2] 
print(Array2) 

, результат:

[[ 0.   0.4284256 0.38090856] 
[ 0.   0.   0.32901526] 
[ 0.   0.   0.  ]]