I am getting error in python code

Question

import pandas as pd
from scipy.spatial import distance

def main():
k = int(input("Enter number of clusters "))

data = {
'X1': [1.1, 1.5, 3.1, 5.1, 3.5, 4.5, 3.5],
'X2': [1.1, 2.1, 4.1, 7.1, 5.1, 5.1, 4.5],
}

dataframe = (pd.DataFrame(data=data)).sample(n=k)
values = dataframe.values.tolist()
cluster = dataframe.values.tolist()
print(cluster)
mean = [[], [], []]

while True:
for i in range(0, k):
for j in range(0, len(values)):
mean[i].append(round(distance.euclidean(cluster[i], values[j]), 2))
if mean==mean:
break

if __name__ == "__main__":
main()

Answer 1

Your code gives an error when we input k > 3, because mean has only 3 elements