Advertisement
Guest User

Untitled

a guest
Mar 18th, 2019
70
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.81 KB | None | 0 0
  1. import pandas
  2.  
  3.  
  4. def map_to_json(user_rates_with_movies):
  5.     genres_list = list(set([item for sublist in user_rates_with_movies['genres'].tolist() for item in sublist]))
  6.     json_columns_list = [('genre-' + item for item in genres_list)]
  7.     json_columns_list.extend(['userID', 'movieID', 'rating'])
  8.     new_data_frame = pandas.DataFrame(columns=json_columns_list)
  9.  
  10.     for row in user_rates_with_movies[['userID', 'movieID', 'rating', 'genres']].to_dict(orient='records'):
  11.         element = {'userID': row['userID'], 'movieID': row['movieID'], 'rating': row['rating']}
  12.  
  13.         for genre in genres_list:
  14.             element[('genre-' + genre)] = ((genre in row['genres']) if int(1) else int(0))
  15.  
  16.         new_data_frame.append(other=element, ignore_index=True)
  17.         print(element)
  18.  
  19.     # print(new_data_frame)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement