Python“最短”挑战（12.26）

Description

输入两个整数类型的矩阵mat1（m行d列）和mat2（d行n列），返回矩阵相乘后的结果mat1*mat2（m行n列）。矩阵均用二维列表进行表示。

Input

两个矩阵分别严格满足m $\times$d 和 d $\times$n的形状（m>=1, d>=1, n>=1，具体数值需要根据输入确定），矩阵中的元素均为整数。

Output

返回相乘后的矩阵，用二维列表表示，每一个元素均为整数
其余要求同首题。

Sample Input

[[1,2]]
[[1],[2]]

Sample Output

[[5]]

Reference Code

def main(mat1,mat2):return [[sum(mat1[i][s]*mat2[s][j] for s in range(len(mat1[0]))) for j in range(len(mat2[0]))] for i in range(len(mat1))] 
print(main(eval(input()),eval(input())))