如何将matlab代码转换成python代码？

Question

下一个代码是matlab代码。我想把这段代码转换成python代码，以便在Arcgis中使用它。在

vals = zeros(size(a(:,:,1)));
 [~,indexOfMax] = max(a,[],3);
 for i=1:size(a,1)
    for j=1:size(a,2)
        vals(i,j) = b(i,j, indexOfMax(i,j));
    end 
 end

我来简单解释一下这个代码。我有两个图像（分别命名为A和B）。两者都有7层相同的尺寸（41696289,7）。首先找到图像中最大值的位置，然后在上一步从图像中提取的最大值位置得到B图像的值。在

非常感谢

Answer 1

vals = numpy.zeros((a.shape[0], a.shape[1]))
indexOfMax = numpy.argmax(a,2)
for i in range(a.shape[0]):
  for j in range(a.shape[1]):
    vals[i,j] = b[i,j, indexOfMax[i,j]]

也应该可以向量化，而不是使用for循环。在

小心，在Python中缩进很重要！保留第二个for循环之前和vals之前的缩进。在