Поиск кластеров чисел в списке
Я борюсь с этим, так как я уверен, что дюжина циклов for не является решением этой проблемы:
Есть отсортированный список номеров, таких как
numbers = [123, 124, 128, 160, 167, 213, 215, 230, 245, 255, 257, 400, 401, 402, 430]
и я хочу создать диктат со списками чисел, в котором разность чисел (следующих друг за другом) не превышает 15. Таким образом, результат будет следующим:
clusters = {
1 : [123, 124, 128],
2 : [160, 167],
3 : [213, 215, 230, 245, 255, 257],
4 : [400, 401, 402],
5 : [430]
}
Мое текущее решение немного некрасиво (я должен удалить дубликаты в конце ...), я уверен, что это может быть сделано питоническим способом.
Это то, что я делаю сейчас:
clusters = {}
dIndex = 0
for i in range(len(numbers)-1) :
if numbers[i+1] - numbers[i] <= 15 :
if not clusters.has_key(dIndex) : clusters[dIndex] = []
clusters[dIndex].append(numbers[i])
clusters[dIndex].append(numbers[i+1])
else : dIndex += 1