как сгруппировать близкие числа из набора?

Question

Для примера рассмотрим множество:

a = [123,234,345,456,124,568,10000,15000,564] 

Затем я нужно сгруппировать вышеуказанные цифры в наборе как

a = [123,234,345,456,124,568,564] 

Пример 2:

a = [10000,12345,11000,10,5000,10500,13000] 

Затем я нужно сгруппировать в выше числа в наборе в качестве

a = [10000,12345,11000,10500,13000] 

поэтому базовая концепция состоит в том, чтобы сгруппировать элементы, которые близки друг к другу.so, пожалуйста, предложите общий код python для этого?

Answer 1

Если вы уже знаете, сколько кластеров у вас есть, я предлагаю использовать K-Means кластеризацию из scikit-learn пакета:

from sklearn.cluster import KMeans 
kmeans = KMeans(n_clusters=2) # numbers will be clustered into 2 clusters based on euclidean distance 
numbers = [123,234,345,456,124,568,10000,15000,564] 
X = [[x] for x in numbers] 
kmeans.fit(X) 
zip(numbers, kmeans.predict(X)) 

Выход (первый элемент является числом, второй кластер идентификатор назначен):

[(123, 1), 
(234, 1), 
(345, 1), 
(456, 1), 
(124, 1), 
(568, 1), 
(10000, 0), 
(15000, 0), 
(564, 1)] 

Или для второго примера:

[(10000, 0), 
(12345, 0), 
(11000, 0), 
(10, 1), 
(5000, 1), 
(10500, 0), 
(13000, 0)] 

Существует ряд других методов, таких как выбор оптимального количества кластеров, но я боюсь, что это немного выходит за рамки. Надеюсь, это поможет.