Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas
- def map_to_json(user_rates_with_movies):
- genres_list = list(set([item for sublist in user_rates_with_movies['genres'].tolist() for item in sublist]))
- json_columns_list = [('genre-' + item for item in genres_list)]
- json_columns_list.extend(['userID', 'movieID', 'rating'])
- new_data_frame = pandas.DataFrame(columns=json_columns_list)
- for row in user_rates_with_movies[['userID', 'movieID', 'rating', 'genres']].to_dict(orient='records'):
- element = {'userID': row['userID'], 'movieID': row['movieID'], 'rating': row['rating']}
- for genre in genres_list:
- element[('genre-' + genre)] = ((genre in row['genres']) if int(1) else int(0))
- new_data_frame.append(other=element, ignore_index=True)
- print(element)
- # print(new_data_frame)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement