python - 如何从包含 NaN 的数组中提取连续元素

Question

考虑以下 numpy 数组

x = np.array([1, 2, np.nan, np.nan, 3, 4, 5, np.nan])

我想提取 x 中的所有非 NaN 连续元素，预期的输出是列表

y = [[1, 2],[3, 4, 5]]

他们的 any 方法是否比简单的 for 循环既优雅又快速？

Answer 1

使用 itertools.groupby

from itertools import groupby

result = [list(map(int,g)) for k,g in groupby(x, np.isnan) if not k]
print (result)
#[[1, 2], [3, 4, 5]]