Files

6 lines
182 B
Python

a=[[1,2],[3,4]]
b=[[5],[6]]
assert all(len(row)==len(b) for row in a)
out=[[sum(a[i][k]*b[k][j] for k in range(len(b))) for j in range(len(b[0]))] for i in range(len(a))]
print(out)