У меня есть задача объединить два pandas кадра данных.
1. https://github.com/timyitong/mf.recommendation/blob/master/data/hetrec2011-movielens-2k-v2/user_ratedmovies.dat
2. https://github.com/timyitong/mf.recommendation/blob/master/data/hetrec2011-movielens-2k-v2/movie_genres.dat
mov ie с идентификатором 3 имеет 2 жанра
Задача состоит в создании кадра данных со столбцами: userID movieID rating genre-Action genre-Animation ... genre-Western. Mov ie с ID 3 должен иметь значение 1 в столбцах genre-Comedy и genre-Romance. Мне удалось вырезать только ненужные столбцы и назначить значения по умолчанию для столбцов нового жанра.
import pandas as pd
mv_data = pd.read_table("movie_genres.dat")
ur_data = pd.read_table("user_ratedmovies.dat", usecols=['userID', 'movieID', 'rating'])
ur_data['genre-Action'] = 0
ur_data['genre-Adventure'] = 0
ur_data['genre-Animation'] = 0
ur_data['genre-Children'] = 0
ur_data['genre-Comedy'] = 0
ur_data['genre-Crime'] = 0
ur_data['genre-Documentary'] = 0
ur_data['genre-Drama'] = 0
ur_data['genre-Fantasy'] = 0
ur_data['genre-Film-Noir'] = 0
ur_data['genre-Horror'] = 0
ur_data['genre-IMAX'] = 0
ur_data['genre-Musical'] = 0
ur_data['genre-Mystery'] = 0
ur_data['genre-Romance'] = 0
ur_data['genre-Sci-Fi'] = 0
ur_data['genre-Short'] = 0
ur_data['genre-Thiller'] = 0
ur_data['genre-War'] = 0
ur_data['genre-Western'] = 0
print(ur_data)
print(mv_data)
выход