Problem

8 /12


対角配列

Theory Click to read/hide

対角配列

 diag(V, k=0) 関数を使用すると、配列から対角線を抽出したり、1 次元配列から対角線配列を作成したりできます。
V - 配列のようなオブジェクト、2 次元または 1 次元の配列、行列、リスト、またはタプル、またはリストまたはタプルを返すメソッドを持つ関数またはオブジェクト。
k - 対角線のインデックス (オプション)。
デフォルトは k = 0 で、これは主対角線に対応します。正の k 値は対角線を上に移動し、負の値は対角線を下に移動します。

この関数は、array NumPy (ndarray) - 指定された配列対角線または指定された 1 次元配列からの対角線配列を返します。

Problem

入力は数値 n です。サイズ nxn の配列を出力します。 in in 対角線の数字は 0 から  n-1 で、残りの数字は 0 です。 .
 

 

<頭> <本体>
# 入力 出力
1 5 [[0 0 0 0 0]
 [0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 0 4]]