Guest User

Untitled

a guest
Jun 25th, 2018
88
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.87 KB | None | 0 0
  1. def plot_dendogram(df_one_race, filename):
  2.  
  3. # Data prep
  4. df = df_one_race.drop(index_list, axis=1)
  5. X = df.as_matrix(columns=None)
  6. Z = linkage(X, 'ward')
  7.  
  8. # call dendrogram to get the returned dictionary (plotting parameters can be ignored at this point)
  9. R = dendrogram(
  10. Z,
  11. no_plot=True,
  12. )
  13.  
  14. # create a label dictionary
  15. labels = df_one_race.driverRef
  16. temp = {R["leaves"][ii]: labels[ii] for ii in range(len(R["leaves"]))}
  17. def llf(xx):
  18. return "{}".format(temp[xx])
  19.  
  20. dendrogram(
  21. Z,
  22. leaf_label_func=llf,
  23. leaf_rotation=90.,
  24. leaf_font_size=8.,
  25. )
  26.  
  27. plt.title('Dendrogram')
  28. plt.xlabel('Drivers')
  29. plt.ylabel('Distance')
  30. plt.savefig(directory+filename, bbox_inches='tight', facecolor='w', transparent=False, pad_inches = 0.5)
  31. plt.show()
Add Comment
Please, Sign In to add comment