Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Numpy를 사용하여 단위 행렬을 만드는 방법은 무엇입니까?

<시간/>

이 프로그램에서 우리는 크기가 nxn인 단위 행렬을 출력할 것입니다. 여기서 n은 사용자의 입력으로 사용됩니다. 요소의 차원과 데이터 유형을 매개변수로 사용하는 numpy 라이브러리에서 identity() 함수를 사용할 것입니다.

알고리즘

1단계:numpy를 가져옵니다. 2단계:사용자의 입력으로 차원을 가져옵니다. 3단계:numpy.identity() 함수를 사용하여 단위 행렬을 인쇄합니다.

예시 코드

numpy를 npdimension =int(input("식별 행렬의 차원 입력:"))identity_matrix =np.identity(dimension, dtype="int")print(identity_matrix)로 가져오기

출력

식별 행렬의 차원을 입력하십시오. 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]]