矩陣使用python

Question

我是新來的python，我正在寫一個程序來矩陣，但有一個問題，我不知道得到正確的輸出，我需要幫助它。 這是一個問題：給定一個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

您的功能：

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 

這似乎是一樣this function。

如果我運行此：

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

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

它返回：

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

這是一樣的numpy的：

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]] 

與同爲矩陣乘法更簡潔地說：

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庫來解決您的問題。

進口numpy的作爲NP

X = np.array（（（2,3），（3,5）））

Y = np.array（（（1,2） ，（5，-1）））

打印X * Y

陣列（[[2,6]， [15，-5]]）

個更多的例子： http://www.python-course.eu/matrix_arithmetic.php

下載numpy的： http://scipy.org/Download