Статья Автор: Богатов Георгий

27 задача на кластеры

import numpy as np
import matplotlib.pyplot as plt
File_data = np.loadtxt("27_A_17882.txt", dtype=float)
# , skiprows=1
print(len(File_data))

#x, y = File_data.T
#plt.scatter(x, y)

plt.xticks(np.arange(-2, 4.5, 0.5))
plt.yticks(np.arange(0, 7, 0.5))
plt.grid()

k1, k2 = (-0.5, 1), (3, 4.7)
plt.scatter(k1[0], k1[1], color='green')
plt.scatter(k2[0], k2[1], color='green')

klast1 = np.array([])
klast2 = np.array([])

for i in File_data:
	x, y = i[0], i[1]
	r1 = ((x - k1[0]) ** 2 + (y - k1[1]) ** 2)**0.5
	r2 = ((x - k2[0]) ** 2 + (y - k2[1]) ** 2)**0.5
	if r1 < r2:
		klast1 = np.append(klast1, i)
	else:
		klast2 = np.append(klast2, i)

klast1 = klast1.reshape((len(klast1) // 2, 2))
klast2 = klast2.reshape((len(klast2) // 2, 2))

x, y = klast1.T
plt.scatter(x, y, color='red')
x, y = klast2.T
plt.scatter(x, y, color='blue')
plt.show()

Attached files
27_A_17882.txt
27_B_17882.txt

Загрузка...

Чтобы оставить комментарий, необходимо авторизоваться

💬

Пока нет комментариев. Будьте первым!

Печать