Advertisement
praktikum

Untitled

Dec 9th, 2019
119
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.44 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.  
  14. music_user = df[(df['user_id']==search_id) & (df['total_play_seconds']>0)]
  15.  
  16. sum_music_user = music_user.groupby('genre_name')['total_play_seconds'].sum()
  17. print(sum_music_user)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement