Python：如何在向量中保存循环最大化器参数的索引？

Question

我有这样的代码：

for i_iter in range(0,max_iter):
    for j_now in range(0,400):
         V[j_now] = np.max(Matrix400_400[:,j_now] + beta * V_old)
    V_old = deepcopy(V)

我想在一个名为index的向量中保存最大化器的索引w.r.t Matrix400_400 [：，j_now]。 在Matlab中，这将是这样的：

[V(j_now), index(j_now)] = max(Matrix400_400(:,j_now) + beta*V_old')

但我需要知道这在python中是如何工作的......

如果某人会很棒。可以帮助我！

干杯，托比亚斯