識別NumPy數組中的連續值組

Question

我有一個NumPy值數組，我需要識別數組中的連續值組。

我嘗試編寫一個“ for”循環來執行此操作，但是我遇到了很多問題。 因此，我在itertools中查看了groupby的文檔。 我以前從未使用過它，並且我對文檔感到有些困惑，所以我來了。

有人可以對使用groupby進行更“外行講話”的解釋嗎？ 本質上，我不需要樣本代碼，而只是對文檔的更詳盡的解釋。

Answer 1

一個很好的答案是使用生成器對其進行分組（可能不是最快的方法）

def groupings(a):
     g = []
     for val in a:
         if not g:
             g.append(val)
         elif abs(g[-1] - val) <= 1.00001:
             g.append(val)
         else:
             yield g
             g = []

print list(groupings(my_numpy_array))

我知道這不會給您一個通俗易懂的分組依據說明（符合某些條件的分組連續項目...對於這種類型的應用程序來說會有些痛苦）