对于movielens数据集统计每一个用户对于电影的观看次数得Python代码怎么写
时间: 2024-02-03 11:11:20 浏览: 125
movielens数据和协同滤波python代码
可以使用pandas库来读取movielens数据集,并进行统计每个用户对电影的观看次数。
代码如下:
```python
import pandas as pd
# 读取movielens数据集
ratings = pd.read_csv('path/to/movielens/ratings.csv')
# 统计每个用户对电影的观看次数
user_movie_counts = ratings.groupby('userId')['movieId'].count()
print(user_movie_counts)
```
其中,`groupby`方法可以将数据按照指定的列进行分组,然后使用`count`方法统计每组的数量。这里按照`userId`列进行分组,统计每个用户对电影的观看次数。最后输出的结果是一个Series,其中索引是用户ID,值是观看次数。
阅读全文