Заполнение нескольких диагональных элементов двумерного массива numpy

Question

Каков наилучший способ заполнения нескольких диагональных элементов (но не всех) 2-мерного массива numpy. Я знаю, что numpy.fill_diagonal - это рекомендуемый способ заполнить все диагональные элементы.

В настоящее время я просто с помощью цикла:

for i in a_list_of_indices: a_2d_array[i,i] = num 

Если массив велик и число диагональных элементов, подлежащих заполнению также велико, есть лучший способ, чем выше.

Answer 1

Вы можете использовать это без зацикливания:

a_2d_array[a_list_of_indices,a_list_of_indices] = num 

Пример:

a_2d_array = np.zeros((5,5)) 
a_list_of_indices = [2, 3] 

возвращается:

array([[ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 1., 0., 0.], 
     [ 0., 0., 0., 1., 0.], 
     [ 0., 0., 0., 0., 0.]])