Problem

8 /12


آرایه مورب

Theory Click to read/hide

آرایه های مورب

تابع  diag(V, k=0) به شما امکان می دهد یک مورب از یک آرایه استخراج کنید و همچنین آرایه های مورب را از آرایه های یک بعدی بسازید.
V - یک شی آرایه مانند، آرایه های دو بعدی یا یک بعدی، ماتریس ها، لیست ها یا تاپل ها، یا هر تابع یا شی با روشی که یک لیست یا تاپل را برمی گرداند.
k - شاخص مورب (اختیاری).
پیش فرض k = 0 است که با قطر اصلی مطابقت دارد. یک مقدار مثبت k مورب را به سمت بالا و یک مقدار منفی آن را به پایین می برد.

تابع آرایه NumPy (ndarray) را برمی گرداند - مورب آرایه مشخص شده یا یک آرایه مورب از آرایه یک بعدی مشخص شده.

Problem

ورودی عدد n است. خروجی آرایه ای با اندازه nxn، در در اعداد قطری از 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]]