Problem

8 /12


Susunan Diagonal

Theory Click to read/hide

Susun Serong

Fungsi  diag(V, k=0) membolehkan anda mengekstrak pepenjuru daripada tatasusunan, serta membina tatasusunan pepenjuru daripada tatasusunan satu dimensi.
V - Objek seperti tatasusunan, tatasusunan dua dimensi atau satu dimensi, matriks, senarai atau tupel, atau sebarang fungsi atau objek dengan kaedah yang mengembalikan senarai atau tupel.
k - indeks pepenjuru (pilihan).
Lalai ialah k = 0 yang sepadan dengan pepenjuru utama. Nilai k positif menggerakkan pepenjuru ke atas, nilai negatif menggerakkannya ke bawah.

Fungsi mengembalikan tatasusunan NumPy (ndarray) - pepenjuru tatasusunan yang ditentukan atau tatasusunan pepenjuru daripada tatasusunan satu dimensi yang ditentukan.

Problem

Input ialah nombor n. Keluarkan tatasusunan saiz nxn, dalam dalam nombor pepenjuru ialah 0 hingga n-1 dan nombor selebihnya ialah 0 .
 

 

Contoh

# Input Output
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]]