livised

Untitled

Nov 5th, 2019
145
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.70 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['user_id'] == search_id) & (df['total_play_seconds'] > 0)]
  14. #sum_music_user = music_user[(music_user['genre_name']) & music_user['total_play_seconds'].sum()]
  15. #sum_music_user = music_user.sort_values(by = 'genre_name')
  16. #sum_music_user = ['total_play_seconds'].sum()
  17. sum_music_user = music_user.groupby(['genre_name', 'total_play_seconds']).sum()
  18. #df.groupby(['Fruit','Name']).sum()
  19. print(sum_music_user)
Advertisement
Add Comment
Please, Sign In to add comment