海报图像读取

电影发布时，都会包含电影海报，海报图像的名字以"mov_id" + 电影ID + ".jpg"的方式命名。因此，我们可以用电影ID去索引对应的海报图像。

海报图像展示如下：

海报图像读取 - 图1

图1： 电影ID-2296的海报

海报图像读取 - 图2

图2： 电影ID-2291的海报

我们可以从新的评分数据文件 new_rating.txt 中获取到电影ID，进而索引到图像，实现如下：

  1. from PIL import Image
  2. import matplotlib.pyplot as plt
  4. # 使用海报图像和不使用海报图像的文件路径不同，处理方式相同
  5. use_poster = True
  6. if use_poster:
  7.     rating_path = "./work/ml-1m/new_rating.txt"
  8. else:
  9.     rating_path = "./work/ml-1m/ratings.dat"
  11. with open(rating_path, 'r') as f:
  12.     data = f.readlines()
  14. # 从新的rating文件中收集所有的电影ID
  15. mov_id_collect = []
  16. for item in data:
  17.     item = item.strip().split("::")
  18.     usr_id,movie_id,score = item[0],item[1],item[2]
  19.     mov_id_collect.append(movie_id)
  22. # 根据电影ID读取图像
  23. poster_path = "./work/ml-1m/posters/"
  25. # 显示mov_id_collect中第几个电影ID的图像
  26. idx = 1
  28. poster = Image.open(poster_path+'mov_id{}.jpg'.format(str(mov_id_collect[idx])))
  30. plt.figure("Image") # 图像窗口名称
  31. plt.imshow(poster)
  32. plt.axis('on') # 关掉坐标轴为 off
  33. plt.title("poster with ID {}".format(mov_id_collect[idx])) # 图像题目
  34. plt.show()

海报图像读取 - 图3

  1. <Figure size 432x288 with 1 Axes>