各pythonの異なる値で対角行列を入力してくださいnumpy

Question

対角要素に同じ値を割り当てる関数numpy.fill_diagonalがあります。しかし、私は対角要素ごとに異なる乱数値を割り当てたい。どのように私はそれをPythonで行うことができますか？ scipyや他のライブラリを使用している可能性がありますか？

Answer 1

np.diag_indicesを使用してそれらのインデックスを取得し、それらのインデックスを使用してインデックスにインデックスを付けるだけで、値を割り当てることができます。

ここでそれを説明するためのサンプル実行だ -

In [86]: arr   # Input array 
Out[86]: 
array([[13, 69, 35, 98, 16], 
     [93, 42, 72, 51, 65], 
     [51, 33, 96, 43, 53], 
     [15, 26, 16, 17, 52], 
     [31, 54, 29, 95, 80]]) 

# Get row, col indices 
In [87]: row,col = np.diag_indices(arr.shape[0]) 

# Assign values, let's say from an array to illustrate 
In [88]: arr[row,col] = np.array([100,200,300,400,500]) 

In [89]: arr 
Out[89]: 
array([[100, 69, 35, 98, 16], 
     [ 93, 200, 72, 51, 65], 
     [ 51, 33, 300, 43, 53], 
     [ 15, 26, 16, 400, 52], 
     [ 31, 54, 29, 95, 500]]) 

ます。またnp.diag_indices_fromを使用することができますし、おそらくそうのような、よりidomaticだろう -

row, col = np.diag_indices_from(arr)