Problem

8 /12


Mảng chéo

Theory Click to read/hide

Mảng chéo

Hàm  diag(V, k=0) cho phép bạn trích xuất một đường chéo từ một mảng, cũng như xây dựng các mảng đường chéo từ các mảng một chiều.
V - Một đối tượng dạng mảng, mảng hai chiều hoặc một chiều, ma trận, danh sách hoặc bộ hoặc bất kỳ hàm hoặc đối tượng nào có phương thức trả về danh sách hoặc bộ.
k - chỉ số của đường chéo (tùy chọn).
Giá trị mặc định là k = 0 tương ứng với đường chéo chính. Giá trị k dương sẽ di chuyển đường chéo lên trên, giá trị âm sẽ di chuyển nó xuống dưới.

Hàm trả về mảng NumPy (ndarray) - đường chéo mảng đã chỉ định hoặc một mảng chéo từ mảng một chiều đã chỉ định.

Problem

Đầu vào là số n. Xuất một mảng có kích thước nxn, in in các số đường chéo là 0 đến n-1 và các số còn lại là 0 .
 

 

Ví dụ
<đầu>
# Đầu vào Đầu ra
1 5 [[0 0 0 0]
 [0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 0 4]]