aykhans/movier
1
0
Fork 0
mirror of https://github.com/aykhans/movier.git synced 2025-12-16 02:29:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

First commit

Browse Source
This commit is contained in:
Aykhan
2023-08-06 22:48:47 +04:00
commit b60673ad25
5 changed files with 879 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
main.py Normal file
View File

@@ -0,0 +1,23 @@
from IMDB_DTO import DTO
from time import time
import pandas as pd
if __name__ == '__main__':
start_time = time()
dto = DTO()
tconst_list = dto.filter_tconst(name='title.basics.tsv')
dto.df2csv(df=pd.DataFrame({'tconst': tconst_list}), name='tconst.csv')
tconst_list = dto.get_tconst('tconst.csv')
df = dto.filter_basic(name='title.basics.tsv', tconst_list=tconst_list)
dto.df2csv(df=df, name='basics.csv')
del df
df = dto.filter_principal(name='title.principals.tsv', tconst_list=tconst_list)
dto.df2csv(df=df, name='principals_comma.csv', overwrite=True)
del df
df = dto.filter_rating(name='title.ratings.tsv', tconst_list=tconst_list)
dto.df2csv(df=df, name='ratings.csv', overwrite=1)
del df