Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để tạo một ma trận nhận dạng bằng Numpy?

Trong chương trình này, chúng ta sẽ in một ma trận nhận dạng có kích thước nxn trong đó n sẽ được lấy làm đầu vào từ người dùng. Chúng tôi sẽ sử dụng hàm nhận dạng () trong thư viện numpy lấy thứ nguyên và kiểu dữ liệu của các phần tử làm tham số

Thuật toán

 Bước 1:Nhập mã số. Bước 2:Lấy kích thước làm đầu vào từ người dùng. Bước 3:In ma trận nhận dạng bằng hàm numpy.identity (). 

Mã mẫu

 nhập numpy dưới dạng npdimension =int (input ("Nhập kích thước của ma trận nhận dạng:")) ID_matrix =np.identity (kích thước, dtype ="int") print (ID_matrix) 

Đầu ra

 Nhập kích thước của ma trận đồng nhất: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]]