Pandas pandas pandas

Question

Has anyone done the 'pandas pandas pandas' question at the end of the data science course. I passed all the cases except case 3 which is locked.

Answer 1

Kindly show your attempt

Answer 2

import numpy as np
import math
n = int(input())
points = []
while n > 0:
points.append( [float(i) for i in input().split()])
n -= 1
cluster1 = []
cluster2 = []
c1 = [0,0]
c2 = [2,2]
for point in points:
x3 = np.array(c2)
x1 = np.array(c1)
x2 = np.array(point)
dist1 =np.sqrt(((x1-x2)**2).sum())
dist2 = np.sqrt(((x3-x2)**2).sum())
if dist1 <= dist2 :
cluster1.append(point)
else:
cluster2.append(point)
if cluster1 == []:
print(None )
if cluster2 == []:
print(None )
result = np.array(cluster1)
result2 = np.array(cluster2)
avg = np.average(result , axis=0)
avg2 = np.average(result2 , axis=0)
#print(cluster1[0])
avg = np.round(avg, decimals =2)
avg2 = np.round(avg2, decimals =2)
print (avg)
print (avg2)