파이썬을 사용하는 행렬

Question

저는 파이썬에 익숙하지 않고 프로그램을 작성하려고합니다.하지만 올바른 결과물을 얻지 못하고 문제가 있습니다. 도움이 필요합니다. 이것은 질문입니다 : 주어진 nXn 행렬 A와 kXn 행렬 B는 AB를 찾습니다. 여기까지 내가 지금까지 가지고있는 것입니다. 미리 감사드립니다.

def matrixmult (A, B): 
    rows_A = len(A) 
    cols_A = len(A[0]) 
    rows_B = len(B) 
    cols_B = len(B[0]) 

    if cols_A != rows_B: 
     print "Cannot multiply the two matrices. Incorrect dimensions." 
     return 

    # Create the result matrix 
    # Dimensions would be rows_A x cols_B 
    C = [[0 for row in range(cols_B)] for col in range(rows_A)] 
    print C 

    for i in range(rows_A): 
     for j in range(cols_B): 
      for k in range(cols_A): 
       C[i][j] += A[i][k]*B[k][j] 
    return C 

Answer 1

함수 :

this function와 동일하게 나타납니다

def matrixmult (A, B): 
    rows_A = len(A) 
    cols_A = len(A[0]) 
    rows_B = len(B) 
    cols_B = len(B[0]) 

    if cols_A != rows_B: 
     print "Cannot multiply the two matrices. Incorrect dimensions." 
     return 

    # Create the result matrix 
    # Dimensions would be rows_A x cols_B 
    C = [[0 for row in range(cols_B)] for col in range(rows_A)] 
    print C 

    for i in range(rows_A): 
     for j in range(cols_B): 
      for k in range(cols_A): 
       C[i][j] += A[i][k]*B[k][j] 
    return C 

.

나는 실행하는 경우이 :

matrix=[[1,2,3], 
    [4,5,6], 
    [7,8,9]] 

print matrixmult(matrix, matrix) # that is your function... 

그것은 반환 행렬 곱셈 더 간결하게 언급 한 바와 같이

import numpy as np 

a=np.array(matrix) 
b=np.array(matrix) 
print np.dot(a,b) 
# [[ 30 36 42] 
    [ 66 81 96] 
    [102 126 150]] 

그리고 같은 :

[[30, 36, 42], [66, 81, 96], [102, 126, 150]] 

이 NumPy와 동일하다 :

def mult(mtx_a,mtx_b): 
    tpos_b = zip(*mtx_b) 
    rtn = [[ sum(ea*eb for ea,eb in zip(a,b)) for b in tpos_b] for a in mtx_a] 
    return rtn 

아마도 입력 데이터가 문제 일 것입니다.

Answer 2

numPy 라이브러리를 사용하여 문제를 해결하십시오. NP로서

수입 NumPy와

np.array X = (((2,3), (3, 5)))

np.array Y = (((1,2) (5, -1)))

인쇄의 X * y를

어레이 ([2,6] [15, -5])

더 많은 예제 : http://www.python-course.eu/matrix_arithmetic.php

다운로드 NumPy와 : http://scipy.org/Download