Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from pandas import DataFrame
- import matplotlib.pyplot as plt
- from sklearn.cluster import KMeans
- import tkinter as tk
- from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
- Data = {'x': [25,34,22,27,33,33,31,22,35,34,67,54,57,43,50,57,59,52,65,47,49,48,35,33,44,45,38,43,51,46],
- 'y': [79,51,53,78,59,74,73,57,69,75,51,32,40,47,53,36,35,58,59,50,25,20,14,12,20,5,29,27,8,7]
- }
- df = DataFrame(Data,columns=['x','y'])
- kmeans = KMeans(n_clusters=3).fit(df)
- centroids = kmeans.cluster_centers_
- root= tk.Tk()
- canvas1 = tk.Canvas(root, width = 100, height = 100)
- canvas1.pack()
- label1 = tk.Label(root, text=centroids, justify = 'center')
- canvas1.create_window(70, 50, window=label1)
- figure1 = plt.Figure(figsize=(5,4), dpi=100)
- ax1 = figure1.add_subplot(111)
- ax1.scatter(df['x'], df['y'], c= kmeans.labels_.astype(float), s=50, alpha=0.5)
- ax1.scatter(centroids[:, 0], centroids[:, 1], c='red', s=50)
- scatter1 = FigureCanvasTkAgg(figure1, root)
- scatter1.get_tk_widget().pack(side=tk.LEFT, fill=tk.BOTH)
- root.mainloop()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement