如何處理在foor循環中大小不等的數組

Question

在我的代碼中，有一個矩陣可以動態增加大小。在MATLAB中的僞代碼如下：

cnt = 0 
for ii = 1:M 
    for jj = 1:N 
     if (condition satisfied) 
      cnt = cnt + 1 
      A(cnt, :, :) = I # I is a matrix that is created within the loop 
     end 
    end 
end 

如何使用NumPy的實現這在Python？

Answer 1

import numpy as np 

A = list() 
for i in range(M): 
    for j in range(N): 
     if condition satisfied: 
      A.append(I) # I is a ndarray created within the loop. 

A = np.array(A)