Viết hàm số UnitMatrix(n) với n là số tự nhiên cho trước, hàm trả lại giá trị là ma trận

Giải Tin học 11 Bài 17: Dữ liệu mảng một chiều và hai chiều

Vận dụng 1 trang 85 Tin học 11: Viết hàm số UnitMatrix(n) với n là số tự nhiên cho trước, hàm trả lại giá trị là ma trận bậc n như Hình 17.1.

Lời giải:

def UnitMatrix(n):

 matrix = []

 for i in range(n):

  row = []

  for j in range(n):

   if i == j:

    row.append(1)

   else:

    row.append(0)

  matrix.append(row)

 return matrix