Advertisement
Guest User

Untitled

a guest
Jun 26th, 2019
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.71 KB | None | 0 0
  1. import pandas as pd
  2. df = pd.read_csv('music_log_upd.csv')
  3.  
  4. genre_grouping = df.groupby('user_id')['genre_name']
  5.  
  6. def user_genres(group):
  7. for col in group:
  8. if len(col[1]) > 50:
  9. user = col[0]
  10. return user
  11.  
  12. search_id = user_genres(genre_grouping)
  13. music_user = df[(df['total_play_seconds']>0) & (df['user_id'] == search_id)]
  14. sum_music_user = music_user.groupby('genre_name')['total_play_seconds'].sum()
  15. #print(sum_music_user)
  16. count_music_user = music_user.groupby('genre_name')['genre_name'].count()
  17. #print(count_music_user)
  18. final_sum = sum_music_user.sort_values(ascending=False)
  19. #print(final_sum)
  20. final_count = count_music_user.sort_values(ascending=False)
  21. print(final_count)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement