如何在每个循环中遍历不同数量的元素？

Question

我有一个三叉树模型，其股价如下图所示。我想编写一个循环以遍历该列中的元素。我将非零值展平到树中的1D数组中，以便稍后进行scipy优化。但是，在我整理数据之后，我丢失了列索引（时间步长）。我只知道第一个元素来自第1列，第2列到第2列，第4列来自第3列，第5列到13列。 但是以下优化过程仍然需要逐列循环。

如何编写这样的循环： 在第一个循环中，检查第一个元素（第一个），第二个循环，接下来的3个元素（第二个至第4个），第三个循环：接下来的9个元素（第5个至第13个）

Answer 1

伪代码看起来像这样（假设上面的数组称为v）。

startIndex = 0
endIndex = 0
for i = 1 to len(v):
    v[i] // this is the i-th row in your screenshot above.
    v[i][startIndex:endIndex + 1] // slice v[i].

    // update start and end indices
    startIndex = endIndex + 1
    endIndex = endIndex + i * i