有两个文件分别是r1.csv、m1.csv,分别表示电影的评分以及电影的类型,某同学想知道评分4.0以上的电影有哪些,请你帮助他完成这个需求找到好看的电影。 m1.csv文件中字段分别为: movieId(电影编号)、title(电影名称)、genres(题材) r1.csv文件中字段分别为: userId(用户编号)movieId(电影编号)rating(评分)timestamp(时间戳)
时间: 2023-06-09 09:03:54 浏览: 289
您好,根据您的需求,您可以按照以下步骤操作:
1. 首先将两个文件加载到程序中,并将它们合并成一个数据集,通过电影的编号将它们进行关联,这可以使用 pandas 库来实现。假设您已经将这两个文件放在了程序同级路径下,您可以使用以下代码加载数据并进行合并:
import pandas as pd
# 读取 r1.csv 和 m1.csv 文件
ratings = pd.read_csv("r1.csv")
movies = pd.read_csv("m1.csv")
# 将电影数据和评分数据合并到一起
movie_ratings = pd.merge(ratings, movies, on='movieId')
2. 接下来,您可以通过筛选评分大于等于4.0的电影来找到好看的电影,这可以使用 pandas 中的查询语句来实现。以下是一个查询评分大于等于4.0的电影的代码示例:
good_movies = movie_ratings.query('rating >= 4.0')
3. 最后,您可以将找到的好看电影的电影名称打印出来,如下所示:
for title in good_movies['title']:
print(title)
希望这个回答对您有帮助。
阅读全文
相关推荐







