Python基础-矩阵相关操作（未调用re模块）

import random


# 生成一个m行n列的随机矩阵
def GenerateMatrix(m, n):
    lst = []
    for i in range(m):
        line = []
        for j in range(n):
            num = random.randint(1, 21)
            line.append(num)
        lst.append(line)
    return lst


# A=GenerateMatrix(3,3)
# print(A)

# 列表推导式生成一个p行q列的随机矩阵(若初始化矩阵，则random.randint(0,0))
def GenerateMatrix2(p, q):
    return [[random.randint(0, 10) for j in range(q)] for i in range(p)]


# B=GenerateMatrix2(4,4)
# print(B)

# 矩阵相乘(M1为p*q阶矩阵，M2为q*r阶矩阵)
def Multiply(M1, M2, p, q, r):
    lst = [[random.randint(0, 0) for y in range(r)] for x in range(p)]
    k = 0
    while k != r:
        for i in range(p):
            Sum = 0
            for j in range(q):
                Sum += M1[i][j] * M2[j][k]
            lst[i][k] = Sum
        k += 1
    return lst


# A = [[1, 2, 3], [4, 5, 6], [7, 8, 9], [2, 3, 4]]  # 4*3矩阵
# B = [[1, 2], [3, 4], [5, 6]]  # 3*2矩阵
# C = Multiply(A, B, 4, 3, 2)
# print(C)