finished_code.py 533 B

12345678910111213141516171819202122232425262728
  1. import pandas as pd
  2. from matplotlib import pyplot as plt
  3. plt.style.use('seaborn')
  4. data = pd.read_csv('2019-05-31-data.csv')
  5. view_count = data['view_count']
  6. likes = data['likes']
  7. ratio = data['ratio']
  8. plt.scatter(view_count, likes, c=ratio, cmap='summer',
  9. edgecolor='black', linewidth=1, alpha=0.75)
  10. cbar = plt.colorbar()
  11. cbar.set_label('Like/Dislike Ratio')
  12. plt.xscale('log')
  13. plt.yscale('log')
  14. plt.title('Trending YouTube Videos')
  15. plt.xlabel('View Count')
  16. plt.ylabel('Total Likes')
  17. plt.tight_layout()
  18. plt.show()