27.05 Кластеры в трехмерном пространстве

Для начала визуально оценим данные в условии кластеры. Для этого откроем предложенные файлы в , перейдем в раздел «Вставка Диаграммы Точечная». Таким способом можно построим по отдельности двухмерные проекции кластеров на оси , и .

Диаграммы для файла А имеют вид:

Из графика видно, что один кластер имеет ординаты, большие 20, а другой – меньшие 20. Пользуясь этим знанием, разделим данные на кластеры в программе.

Код для файла А

from math import *
f = open(’3_A.txt’)
n = f.readline()
a = [list(map(float,i.replace(’,’,’.’).split())) for i in f]
clusters = [[] for i in range(2)]
for i in a:
    if i[1] > 20:
        clusters[0].append(i)
    else:
        clusters[1].append(i)
sum_x = sum_y = sum_z = tx = ty = tz = 0
for cluster in clusters:
    mn = 10**20
    for star_1 in cluster:
        sm = 0
        for star_2 in cluster:
            sm += dist(star_1,star_2)
        if sm < mn:
            mn = sm
            tx,ty,tz = star_1
    sum_x += tx
    sum_y += ty
    sum_z += tz
print(int(abs(sum_x/2)*10))
print(int(abs(sum_y/2)*10))
print(int(abs(sum_z/2)*10))

Диаграммы для файла Б имеют вид:

Из графика видно, что все кластеры распределены по четырем координатным осям. Пользуясь этим знанием, разделим данные на кластеры в программе.

Код для файла Б

from math import *
f = open(’3_B.txt’)
n = f.readline()

clusters = [[] for i in range(4)]
for i in f:
    star = list(map(float, i.replace(’,’, ’.’).split()))
    if star[0] > 0 and star[1] > 0:
        clusters[0].append(star)
    elif star[0] < 0 and star[1]>0:
        clusters[1].append(star)
    elif star[0]<0 and star[1] < 0:
        clusters[2].append(star)
    else:
        clusters[3].append(star)
sum_x = sum_y = sum_z = tx = ty = tz = 0
for cluster in clusters:
    mn = 10**20
    for star_1 in cluster:
        sm = 0
        for star_2 in cluster:
            sm += dist(star_1,star_2)
        if sm < mn:
            mn = sm
            tx,ty,tz = star_1
    sum_x += tx
    sum_y += ty
    sum_z += tz
print(int(abs(sum_x/4)*10))
print(int(abs(sum_y/4)*10))
print(int(abs(sum_z/4)*10))