1. 程式人生 > >Python“最短”挑戰(12.26)

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())))