>  기사  >  백엔드 개발  >  다차원 배열을 사용하여 두 행렬을 곱하는 Python 프로그램

다차원 배열을 사용하여 두 행렬을 곱하는 Python 프로그램

WBOY
WBOY앞으로
2023-09-11 17:09:07617검색

다차원 배열을 사용하여 두 행렬을 곱하는 Python 프로그램

행렬은 행과 열로 배열된 숫자의 집합입니다. m개의 행과 n개의 열로 구성된 행렬을 m X n 행렬이라고 하며, m과 n을 차원이라고 합니다. 행렬은 Python에서 목록이나 NumPy 배열을 사용하여 만든 2차원 배열입니다.

일반적으로 행렬 곱셈은 첫 번째 행렬의 행과 두 번째 행렬의 열을 곱하여 수행할 수 있습니다. 여기서 첫 번째 행렬의 열 개수는 두 번째 행렬의 행 개수와 같아야 합니다.

입력 및 출력 시나리오

두 개의 행렬 A와 B가 있다고 가정하고, 이 두 행렬의 크기는 각각 2X3과 3X2입니다. 곱셈 후 결과 행렬은 2개의 행과 1개의 열을 갖게 됩니다.

으아아아

또한 행렬의 요소별 곱셈도 수행할 수 있습니다. 이 경우 두 입력 행렬의 행과 열 개수는 모두 동일해야 합니다.

으아아아

For 루프 사용

중첩된 for 루프를 사용하면 두 행렬에 곱셈을 수행하고 그 결과를 세 번째 행렬에 저장합니다.

이 예에서는 곱셈 결과를 저장하기 위해 모두 0으로 결과 행렬을 초기화합니다.

으아아아

출력

으아아아

첫 번째 행렬(matrix_a)의 행과 열 개수는 3개이고, 두 번째 행렬(matrix_b)의 행과 열 개수는 3개입니다. 이 두 행렬(matrix_a, matrix_b)을 곱한 결과 행렬은 3개의 행과 4개의 열(예: 3X4)을 갖게 됩니다.

여기에서는 numpy.array() 함수를 사용하여 @ 연산자를 사용하여 행렬 곱셈을 간단히 수행할 수 있도록 행렬을 생성합니다.

으아아아

출력

으아아아

곱하기 연산자 @는 Python 3.5+ 버전에서 사용할 수 있습니다. 그렇지 않으면 numpy.dot() 함수를 사용할 수 있습니다.

이 예에서는 (*) 별표 연산자를 사용하여 두 numpy 배열의 요소별 곱셈을 수행합니다.

으아아아

출력

으으으으

위 내용은 다차원 배열을 사용하여 두 행렬을 곱하는 Python 프로그램의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
이 기사는 tutorialspoint.com에서 복제됩니다. 침해가 있는 경우 admin@php.cn으로 문의하시기 바랍니다. 삭제