Question

我已经创建了一个小的Python程序来将两个2乘2的矩阵相乘，但我想知道它是否可以以任何方式进行简化（特别是创建新数组）

python代码如下：

matA=[0]*2
matB=[0]*2
matC=[0]*2
matC[0]=[0]*2
matC[1]=[0]*2
# creating new arrays for multiplying two 2 by 2 matrices
# must be a more simple way


def multiply2by2matrices(a,b):

    matC[0][0]=a[0][0]*b[0][0]+a[0][1]*b[1][0]
    matC[0][1]=a[0][0]*b[0][1]+a[0][1]*b[1][1]
    matC[1][0]=a[1][0]*b[0][0]+a[1][1]*b[1][0]
    matC[1][1]=a[1][0]*b[0][1]+a[1][1]*b[1][1]

    print ((matC[0][0]),(matC[0][1]))
    print ((matC[1][0]),(matC[1][1]))

matA[0]=[4,3]
matA[1]=[2,12]

matB[0]=[5,-2]
matB[1]=[6,3]

multiply2by2matrices(matA, matB)

任何想法都将受到极大的欢迎。

Answer 1

不要手动执行。你正在重新发明轮子，那里已经有很好的轮子。

Numpy就是答案。

import numpy as np
a = np.arange(20).reshape(5,4)
b = (np.arange(20) + 10).reshape(4,5)
np.dot(a,b)

文档：

http://docs.scipy.org/doc/numpy/reference/generated/numpy.dot.html

干杯，P

Answer 2

除非你只需要使用vanilla python，否则使用numpy会更简单。您可以使用矩阵类或只使用2d数组和点函数。例如：

import numpy as np
a=np.array([[1,1],[2,2]])
b=np.array([[3,3],[3,3]])
c=np.dot(a,b)

产生

array([[ 6,  6],
       [12, 12]])

Answer 3

你绝对可以简化矩阵乘法并使其一般化。诀窍是使用zip。当然还有其他方法，但我认为zip可能会产生一些更干净的代码。我没有测试以下内容，但我认为我的线性代数对我有用。

def matmult(a,b):
    zipb = zip(*b)
    return [[sum(ax*bx for ax,bx in zip(rowa, colb)) for colb in zipb] for rowa in a]

如果不使用列表推导，则需要预先分配列表或在for循环之间使用append / extend。

Python矩阵查询 - 必须更简单

3 个答案: