用Python实现矩阵相乘

2022年 11月 7日

用Python实现矩阵相乘

题目1：计算两个普通矩阵A和B的乘积

#Description
计算两个矩阵A和B的乘积。
#Input
第一行三个正整数m、p和n，0<=m,n,p<=10，表示矩阵A是m行p列，矩阵B是p行n列；接下来的m行是矩阵A的内容，每行p个整数，用空格隔开；最后的p行是矩阵B的内容，每行n个整数，用空格隔开。
#Output
输出乘积矩阵：输出占m行，每行n个数据，以空格隔开。
#Sample Input
2 3 4

1 0 1
0 0 1

1 1 1 3
4 5 6 7
8 9 1 0

#Sample Output
9 10 2 3
8 9 1 0

#普通矩阵相乘
m,p,n=map(int,input().split(","))#获得两个矩阵的行列数
A=[]
B=[]   #创建A,B两个空列表，用以存放相乘的两个矩阵
C=[[0 for i in range(n)] for j in range(m)] #创建一个m行n列的初始化矩阵
for i in range(m):
    A.append(list(map(int,(input().split(" "))))) #获得A矩阵
for i in range(p):
    B.append(list(map(int,input().split(" ")))) #获得B矩阵
for i in range(m):
    for j in range(n):
        for k in range(p):
            C[i][j] += A[i][k]*B[k][j] #根据矩阵相乘的运算规则进行运算
for s in C:
    print() #保证输出为矩阵形式
    for r in s:
        print(r,end=" ") #输出结果矩阵

题目2：计算两个关系矩阵A和B的乘积

1 0 0
0 0 1
0 1 0
1 0 0

1 0 0
0 1 0
1 0 1
#Sample Output
1 0 0
1 0 1
0 1 0
1 0 0

#关系矩阵相乘，结构与普通矩阵相乘的结构相同
m,p,n=map(int,input().split(","))
A=[]
B=[]
C=[[0 for i in range(n)] for j in range(m)]
for i in range(m):
    A.append(list(map(int,(input().split()))))
for i in range(p):
    B.append(list(map(int,input().split())))
for i in range(m):
    for j in range(n):
        for k in range(p):
            C[i][j] += A[i][k]*B[k][j]
        if C[i][j] >= 1: #我用相加的和与1比较来判断输入1或保持初始化矩阵中该位的值不变（因为初始化矩阵中每位都为0）
            C[i][j] = 1
for s in C:
    print()
    for r in s:
        print(r,end=" ")