python - 如何将矩阵中的每一行乘以 Python 中第二个矩阵中的每一列？

Question

我知道两个矩阵相乘的步骤如下

第 1 步:确保第一个列中的列数等于第二个中的行数。

第 2 步:将第一个矩阵的每一行的元素乘以第二个矩阵的每一列的元素。

第 3 步:添加产品。

第二步怎么做？

例如

A = [[3,4,5],[5,0,6],[5,7,1]]

B = [[2,1,3],[2,6,4]]

到目前为止，我有一个函数可以找到第二列的每一列

def column(B,j):
    col = []
    for column in B:
        col.append(column[j])
    return col

接下来我将不得不创建一个函数来查找第一行的每一行

def rows(A,i):

但是我不知道如何创建一个将它们相乘的函数，就像

row(A,0) • col(B,0)

row(A,0) • col(B,1)

row(A,1) • col(B,0)

row(A,1) • col(B,1)

row(A,2) • col(B,0)

row(A,2) • col(B,1)

Answer 1

你可能应该使用 numpy:

import numpy as np
np.dot(row(A,0), col(B,0))

但是，假设您不想使用它，您可以这样做:

def dot(arr1, arr2):
    return sum([x*y for x,y in zip(arr1, arr2)])

dot(row(A,0), col(B,0))