回到內循環外循環 - Python嵌套循環

Question

我有一個情況，我需要2循環（我和K）如下所示。我想在離開後繼續內循環。

import numpy as np 
X = [[12, 11, 1], [1,2,3]] 
mu = [1, 2, 3] 
sublist = [] 

for i in range(0, 4): 
    for k in range(0, 3): 
     subtr = X[i] - mu[k] 
     sublist.append(subtr) 
    # leaving the loop k to calc argmin 
    agmin = np.argmin(sublist) 
    C.append(agmin) 
     # Now I want to get back to the inner loop (k) to continue #further calculation, but obviously will result an error. 
     np.dot((C[i] == k),X[i])  

處理此類情況的最佳方法是什麼？

Answer 1

在離開內部循環之前，完成內部循環中所需的所有操作。下面是您的代碼的一個略加修改：

import numpy as np 
X = [[12, 11, 1], [1,2,3]] 
mu = [1, 2, 3] 
sublist = [] 
C = #whatever C should be initialized to  

for i in range(0, 4): 
    for k in range(0, 3): 
     subtr = X[i] - mu[k] 
     sublist.append(subtr) 

     # calculate agmin (argmin) once per inner loop, at end 
     if k == 2: 
      agmin = np.argmin(sublist) 
      C.append(agmin) 

     # not sure what this line does, but do it inside the inner loop since it 
     # needs k. (I'm guessing you really want some_var = np.dot(...)) 
     np.dot((C[i] == k),X[i])