Dot and Cross in Python 3 - HackerRank Solution

Dot and Cross

问题描述：

• 给予两个为N×N的矩阵A和B，求它们的矩阵乘积
• 输入格式：
  • 第一行为整数N
  • 接下来的N行为矩阵A中每一行的数据
  • 最后的N行为矩阵B中每一行的数据

样例演示

• 输入：

2
1 2
3 4
1 2
3 4

• 输出：

[[ 7 10 ]
 [ 15 22 ]]

参考代码

import numpy

n = int(input())
a = numpy.array([input().split() for _ in range(n)], int)
b = numpy.array([input().split() for _ in range(n)], int)
print(numpy.dot(a, b))

小结

• input()传回的是字符串，需要用int转换为整型数据
• 多输入考虑使用input().split()来结合使用