如何使用 csv 文件获取 IMDb URL 列表？

Question

我有一个包含 1000 部电影的 .csv 文件（https://github.com/LearnDataSci/articles/blob/master/Python%20Pandas%20Tutorial%20A%20Complete%20Introduction%20for%20Beginners/IMDB-Movie-Data。 csv），并已设置获取所有 1000 部电影的 IMDb URL 的任务。

我尝试过使用 Cinemagoer，但我并不特别想手动插入所有 1000 个影片。有没有更有效的方法？

Answer 1

您可以解析表格并使用“标题”列进行 Cinemagoer 查询。

import requests
import csv
from imdb import Cinemagoer
from io import StringIO

TABLE_URL = 'https://github.com/LearnDataSci/articles/raw/master/Python%20Pandas%20Tutorial%20A%20Complete%20Introduction%20for%20Beginners/IMDB-Movie-Data.csv'


def get_table(url):
    r = requests.get(url, stream=True)

    return StringIO(r.content.decode())


def search_movie(interface, name):
    movies = interface.search_movie(name)

    return movies[0] if len(movies) > 0 else None


def main():
    ia = Cinemagoer()

    reader_obj = csv.reader(
        get_table(TABLE_URL)
    )

    # Skip column names
    next(reader_obj)

    for row in reader_obj:
        # 1 - is column with names
        print(
            search_movie(ia, row[1])
        )


main()