python - 如何使用嵌套for循环将两个矩阵相加？

Question

我是 python 新手，但我有一些 matlab 工作经验。我有一个关于如何将嵌套 for 循环的计算保存到矩阵变量 x(i,j) 的具体问题。

假设x=1:10和y=1:5。现在，如果我需要将 x 的各个元素与 y 的各个元素求和，那么在 matlab 中我有这样的命令

sum=0;

for i=1:length(x)
    for j=1:length(y)
        sum(i,j)=x(i)+y(j);
    end
end

现在我需要将上面的例程转换为Python。怎么可能？

Answer 1

您可以使用列表理解。这将产生与您的 Matlab 代码相同的输出。

x = range(1, 11)
y = range(1, 6)

s = [[xx + yy for yy in y] for xx in x]

print(s)