IMDB的建议

时间:2014-01-31 16:26:16

标签: python api imdb imdbpy

我想制作一种自制的算法,从我所拥有的所有电影列表中,将返回我没见过的电影的准确建议。但为此,我需要知道IMDBpy是否可以返回“建议”部分。 实际上,当您在网站上搜索电影时,您将获得与您搜索的电影类型相匹配的电影列表。

但我无法找到有关IMDBpy文档的答案。有没有办法得到它的建议?

2 个答案:

答案 0 :(得分:2)

Link!

这里有一条线:

def get_movie_recommendations(self, movieID):

也许这可以引导你朝着正确的方向前进。

答案 1 :(得分:1)

由于get_movie_recommendations并不能真正按照您的意愿工作(例如,它不能为12年的奴隶返回任何内容),您可以使用BeautifulSoup获取建议。

import bs4
import imdb
import requests

src = requests.get('http://www.imdb.com/title/tt2024544/').text
bs = bs4.BeautifulSoup(src)
recs = [rec['data-tconst'][2:] for rec in bs.findAll('div', 'rec_item')]
print recs

打印:

['1535109', '0790636', '1853728', '0119217', '2334649', '0095953', '1935179', '2370248', '1817273', '1210166', '0169547', '1907668']

之后,您可以使用IMDBpy ...

搜索这些电影
ia = imdb.IMDb()
for rec in recs:
  movie = ia.get_movie(rec)
  print movie.movieID, movie.get('title')

......输出:

1535109 Captain Phillips
0790636 Dallas Buyers Club
1853728 Django Unchained
0119217 Good Will Hunting
2334649 Fruitvale Station
0095953 Rain Man
1935179 Mud
2370248 Short Term 12
1817273 The Place Beyond the Pines
1210166 Moneyball
0169547 American Beauty
1907668 Flight