解决方法,把二维列表转为一维,代码如下:
import pandas as pd
movie_list = [data]
movie_rating = [data]
uni_id = [data]
movie_pd = pd.DataFrame({'movieid': movie_list,'rating' : movie_rating}, index=uni_id)
movie_pd = movie_pd.fillna(0)
group_movie_id = movie_pd.groupby('movieid')
movie_id_total_rating = group_movie_id.sum()
movie_id_count = group_movie_id.count()
movie_id_aver_rating = movie_id_total_rating / movie_id_count
movie_id_total_rating_lis = [i for item in movie_id_total_rating.values for i in item]
movie_id_count_lis = [i for item in movie_id_count.values for i in item]
movie_id_aver_rating_lis = [i for item in movie_id_aver_rating.values for i in item]
print(movie_id_total_rating_lis)
result_list = pd.DataFrame({'movie_id': movie_id_total_rating.index,
'total_rating': movie_id_total_rating_lis,
'count': movie_id_count_lis,
'aver_rating': movie_id_aver_rating_lis
})
print(result_list)
解决后打印输出ok