PY
py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# solution to Sololearn Python Data Science course Project 4 "Pandas Pandas Pandas"
n = int(input())
X = []
for i in range(n):
X.append([float(x) for x in input().split()])
import numpy as np
X = np.array(X).reshape(-1,2) # 2D array
# define initial centroids as 1D arrays
C1_init = np.array([0, 0])
C2_init = np.array([2, 2])
C1_set = []
C2_set = []
# assign each imput coordinate to either first or second initial centroid
for i in range(n):
# calculate Euclidean distances
dist1 = np.sqrt(((X[i]-C1_init)**2).sum())
dist2 = np.sqrt(((X[i]-C2_init)**2).sum())
if dist1 <= dist2: # compare distances
C1_set.append(X[i])
else:
C2_set.append(X[i])
C1_set = np.array(C1_set).reshape(-1,2) # 2D arrays again
C2_set = np.array(C2_set).reshape(-1,2)
Enter to Rename, Shift+Enter to Preview
OUTPUT
Run