import pandas as pd
df = pd.read_csv('music_log_upd.csv')

genre_grouping = df.groupby('user_id')['genre_name']


def user_genres(group):
  for col in group:
    if len(col[1]) > 50:
      user = col[0]
      return user


search_id = user_genres(genre_grouping)

music_user = df[(df['user_id'] == search_id)&(df['total_play_seconds'] != 0)]

sum_music_user = music_user.groupby('genre_name')['total_play_seconds'].sum()
#print(sum_music_user)
count_music_user = music_user.groupby('genre_name')['genre_name'].count()
print(count_music_user)