Spaces:
Runtime error
Runtime error
import imdb | |
import csv | |
import pandas as pd | |
ia = imdb.IMDb() | |
dict1={} | |
pd_dict={'Action': [], 'Adventure': [], 'Fantasy': [], 'Sci-Fi': [], | |
'Animation': [], 'Comedy': [], 'Family': [], 'Mystery': [], | |
'Romance': [],'Drama': [],'Crime': [], 'Thriller': [], | |
'War': [], 'Musical': [], 'Biography': []} | |
with open("movies.csv", 'r') as file: | |
csvreader = csv.reader(file) | |
j=0 | |
for row in csvreader: | |
if j==250: | |
break | |
if j==0: | |
j=1 | |
continue | |
name=row[1] | |
items = ia.search_movie(name) | |
j+=1 | |
print(j) | |
for i in pd_dict: | |
pd_dict[i].append(0) | |
code = items[0].getID() | |
series = ia.get_movie(code) | |
genre = series.data['genres'] | |
for i in genre: | |
if i in pd_dict: | |
pd_dict[i][-1]=1 | |
print(pd_dict) | |
df=pd.DataFrame(pd_dict) | |
df.to_csv("movies3.csv") | |