Trong chương trình này, in ma trận đơn vị kích thước nxn với n nhập từ người dùng. Dùng identity() trong numpy nhận kích thước và kiểu dữ liệu.
Thuật toán
Bước 1: Import numpy. Bước 2: Nhập kích thước từ người dùng. Bước 3: In ma trận đơn vị bằng numpy.identity().
Mã ví dụ
import numpy as np
dimension = int(input("Nhập kích thước ma trận đơn vị: "))
identity_matrix = np.identity(dimension, dtype="int")
print(identity_matrix)
Kết quả
Nhập kích thước ma trận đơn vị: 5 [[1 0 0 0 0] [0 1 0 0 0] [0 0 1 0 0] [0 0 0 1 0] [0 0 0 0 1]]