結果 : how to write matrix multiplication in python