Pls help in Data Science - Pandas Pandas Pandas

Question

What's the issue in my code

import numpy as np

first = np.array([[0., 0.]])
second = np.array([[2., 2.]])
n = int(input())

data = []

for i in range(n):
data.append([float(i) for i in input().split()])

data = np.array(data).reshape((-1,2))

for i in range(n):
dist1 = np.sqrt(((data[i]-first[0])**2).sum())

dist2 = np.sqrt(((data[i]-second[0])**2).sum())

if (dist1) <= (dist2):
first = np.vstack((first,data[i]))

if first.size > 2:
mean1 = np.mean(first[1:], axis=0)
print(np.around(mean1, decimals=2))
else:
print(None)

if second.size > 2:
mean2 = np.mean(second[1:], axis=0)
print(np.around(mean2, decimals=2))
else:
print(None)

Answer 1

When you are reshaping this array to size (-1,2)
You are reducing the length to half

So when running for loop you have to give range of n//2 and not n

Hope It Helps You ✨

Answer 2

#Here's The Right Code

import numpy as np

first = np.array([[0., 0.]])
second = np.array([[2., 2.]])
n = int(input())

data = []

for i in range(n):
data.append([float(i) for i in input().split()])

data = np.array(data).reshape((-1,2))

for i in range(n//2):
dist1 = np.sqrt(((data[i]-first[0])**2).sum())

dist2 = np.sqrt(((data[i]-second[0])**2).sum())

if (dist1) <= (dist2):
first = np.vstack((first,data[i]))

if first.size > 2:
mean1 = np.mean(first[1:], axis=0)
print(np.around(mean1, decimals=2))
else:
print(None)

if second.size > 2:
mean2 = np.mean(second[1:], axis=0)
print(np.around(mean2, decimals=2))
else:
print(None)