使用 itertools.groupby 返回最長的連續負數列表

Question

鑒於系列：

series = [2, 6, 9, -1, 1, -3, -3, -3, 1,-4]

如何使用itertools.groupby返回列表[-3, -3, -3] ？

試：

from itertools import groupby
max((list(g) for k, g in groupby(series, key=lambda i: i < 0)), key=len)

返回：

[2, 6, 9]

提前致謝

Answer 1

你可以這樣做：

from itertools import groupby

series = [2, 6, 9, -1, 1, -3, -3, -3, 1, -4]
result = max((list(group) for key, group in groupby(series) if key < 0), key=len)

print(result)

Output

[-3, -3, -3]

您的代碼的問題是您在 groupby 中使用了錯誤的鍵key=lambda i: i < 0 ，如果鍵為正，只需過濾結果。

Answer 2

檢查你的k值是真還是假。

from itertools import groupby
series = [2, 6, 9, -1, 1, -3, -3, -3, 1,-4]
x = max([list(g) for k, g in groupby(series, key=lambda i: i < 0) if k == True ], key=len)
print(x)