IIS_2023_1/belyaeva_ekaterina_lab_4/main.py

30 lines
869 B
Python
Raw Permalink Normal View History

2023-11-01 15:55:34 +04:00
import pandas as pd
import matplotlib.pyplot as plt
from scipy.cluster.hierarchy import dendrogram, linkage
# Загрузка данных
data = pd.read_csv('Current_Pub_Meta.csv')
# Выбор нужных столбцов
selected_columns = ['Name', 'Herald Picks', 'Herald Win Rate']
data = data[selected_columns]
# Создание матрицы для анализа
matrix = data.drop('Name', axis=1).values
# Вычисление матрицы связей
linked = linkage(matrix, 'ward')
# Рисование дендрограммы
plt.figure(figsize=(10, 6))
dendrogram(linked,
orientation='top',
labels=data['Name'].tolist(),
distance_sort='descending',
show_leaf_counts=True)
plt.title('Dendrogram of Hero Win Percentage')
plt.xlabel('Heroes')
plt.ylabel('Distance')
plt.xticks(rotation=90)
plt.show()