我正在尝试编写一个函数,添加两个矩阵来传递以下doctests:
>>> a = [[1, 2], [3, 4]]
>>> b = [[2, 2], [2, 2]]
>>> add_matrices(a, b)
[[3, 4], [5, 6]]
>>> c = [[8, 2], [3, 4], [5, 7]]
>>> d = [[3, 2], [9, 2], [10, 12]]
>>> add_matrices(c, d)
[[11, 4], [12, 6], [15, 19]]
Run Code Online (Sandbox Code Playgroud)
所以我写了一个函数:
def add(x, y):
return x + y
Run Code Online (Sandbox Code Playgroud)
然后我写了以下函数:
def add_matrices(c, d):
for i in range(len(c)):
print map(add, c[i], d[i])
Run Code Online (Sandbox Code Playgroud)
我排序的得到正确的答案.