Python中的矩阵乘法
你好(请原谅我的英语),我在使用Python进行矩阵乘法时遇到了一个大问题。我创建了一个列表的列表,然后用一个缩放矩阵去相乘。可是我做的这些似乎在索引上出现了乘法操作的问题。我用纸和笔检查过,结果是对的。难道我在处理索引时做错了,还是一开始就没有正确地安排矩阵?
def main():
if len(sys.argv) > 1:
v = int(sys.argv[1])
else:
print "error python exe:"
print "\tpython <programa.py> <num_vertices>"
A = []
for i in range(v):
A.append([0]*2)
for i in range(v):
for j in range(2):
A[i][j] = input("v: ")
print A
Escala(A)
def Escala(A):
print "Escala"
sx = input("Sx: ")
sy = input("Sy: ")
S = [(sx,0),(0,sy)]
print S
M = mult(S,A)
print M
def mult(m1,m2):
M = zero(len(m1),len(m2[0]))
for i in range(len(m2)):
for j in range(len(m2[0])):
for k in range(len(m1)):
M[i][j] += m1[k][j]*m2[k][j]
print M
return M
def zero(m,n):
# Create zero matrix
new_matrix = [[0 for row in range(n)] for col in range(m)]
return new_matrix
1 个回答
5
我觉得这个有点不对:
M[i][j] += m1[k][j]*m2[k][j]
难道应该是:
M[i][j] += m1[i][k]*m2[k][j]