Jupyter Notebook 循环单元

Question

我有一个问题，是否可以在笔记本中的不同单元格中循环迭代？例如我有这个代码：

for c in range(2):
# Thetas
thetas = np.zeros((J,I))
it = np.nditer(thetas, flags=['multi_index'],op_flags=['writeonly'])
while not it.finished:
    i=it.multi_index[1]
    j=it.multi_index[0]
    coefs=np.array([1-(K/(prod[j]*la[j]*firm[i])),(-K/(prod[j]*firm[i]))*(1+(1/la[j])),-K/(prod[j]*firm[i])])
    sol=np.roots(coefs)
    it[0]=np.amax((sol))
    it.iternext()

# Wages
wag=np.zeros((J,I))
it = np.nditer(wag, flags=['multi_index'],op_flags=['writeonly'])
while not it.finished:
    i=it.multi_index[1]
    j=it.multi_index[0]
    it[0]=prod[j]*firm[i]-(K*(1+thetas[j,i])/(la[j]*thetas[j,i]))
    it.iternext()

出于演示目的，我想在单元格中运行 thetas 代码，显示一些结果，然后运行工资，显示另一个结果，然后对 theta 和工资进行第二次迭代。这是可能吗？

谢谢