遍历numpy数组的“压缩”范围

Question

我经常使用numpy数组来表示时间序列中的关键时刻。 然后，我想遍历这些范围并对其进行操作。 例如：

rngs = [0, 25, 36, 45, ...]
output = []
for left, right in zip(rngs[:-1], rngs[1:]):
      throughput = do_stuff(array[left:right])...
      output.append(throughput)

有没有那么尴尬的方法可以做到这一点？

Answer 1

您可能使用枚举生成器

rngs = [0, 25, 36, 45, ...]
output = []
for index, _ in enumerate(rngs[:-1]):
      throughput = do_stuff(array[index:index+1])...
      output.append(throughput)

与理解列表成一行：

rngs = [0, 25, 36, 45, ...]
output = [do_stuff(array[index:index+1]) for index, _ in enumerate(rngs[:-1])]