Numpy Arrays 基于列表中的阈值标准

Question

我需要一个包含 7000 个元素的列表，值的范围从 ~ -150 到 150，并且它来回循环。 每次该值超过 -50 的阈值时，我都需要将所有值包含在一个数组中，直到它再次超过 -50。 我需要为每个循环创建一个数组，然后计算生成了多少个 arrays，并求出每个 arrays 有多少个元素的平均值。

Answer 1

以下代码 1) 计算下一个元素与 -50 交叉处的索引，2) 在与 -50 交叉的元素处拆分数组，3) 计算拆分后有多少个 arrays，以及 4) 计算每次拆分的平均值。

import numpy as np
array = np.array([...])
crossings = np.where(np.diff(np.sign(array + 50)))[0]
splits = np.split(array, crossings + 1)
count = len(splits)
means = [x.mean() for x in splits]