Python編寫小工具之統計演員票房排行榜

閒來無事想要了解下2018年的票房之王（國產片-內地），因而乎寫了這麼個小工具 先看一下結果：

沒錯，前三是王寶強，張譯，黃景瑜，王寶強榮登票房王寶座！ （票房數據來源時光網）

首先先爬取電影票房數據（時光網）

def getMovieList():
    ''' 2018年內地票房（國產片） '''
    movie_params = {
        'year': '2018',
        'area': 'china',
        'type': 'MovieRankingYear',
        'category': 'china',
        'page': 0,
        'display': 'list',
        'timestamp': '1553422026885',
        'version': '07bb781100018dd58eafc3b35d42686804c6df8d',
        'dataType': 'json'
    }
    url = 'http://movie.mtime.com/boxoffice/'
    movie_list = []
    #總共10頁數據
    for j in range(10):
        movie_params['page'] = j
        r = requests.get(url, params=movie_params).json()
        soup = BeautifulSoup(r['html'], 'lxml')
        movie_div = soup.find('div', {'class': 'boxofficelist'})
        movie_dd = movie_div.find_all('dd')
        for i in range(len(movie_dd)):
            movie_name = movie_dd[i].find('h3').text
            totalnum = movie_dd[i].find_all('p', {'class': 'totalnum'})[0].text
            actors = movie_dd[i].find_all('p')[3].text[3:]
            movie_list.append((movie_name, totalnum, actors))
    return movie_list
複製代碼

將爬取的數據保存到CSV文件中

def write_movie_dsv(movies):

   ''' 將2018內地票房（國產片）列表寫入csv文件中 '''
   header=['電影名','票房','演員']
   with open('movie_2018_chinalist.csv','w',newline='') as csvfile:
       writer=csv.writer(csvfile)
       writer.writerow(header)
       for i,movie in enumerate(movies):
           if (i+1)%10==0:
               print ('已處理{}條記錄（共{}條記錄）'.format(i+1,len(movies)))
           writer.writerow(movie)
複製代碼

統計2018內地（國產片）演員的票房，並寫入csv文件中

def write_actorTotalNum_dsv(movies):

   ''' 統計2018內地（國產片）演員的票房，並寫入csv文件中 '''
   movie_actors = []
   actor_totalnum_temp = []
   actor_totalnum = {}
   header = ['演員', '票房(億)']
   with open('actor_totalnum_2018.csv', 'w', newline='') as csvfile:
       writer = csv.writer(csvfile)
       writer.writerow(header)
       for movie in movies:
           movie_actor = movie[2].split('/')
           for i in range(len(movie_actor)):
               if movie_actor[i] != '':
                   actor_totalnum_temp.append((movie_actor[i], movie[1]))
       for i, j in actor_totalnum_temp:
           if j[-1:] == '萬':
               actor_totalnum[i] = actor_totalnum.get(i, 0.0) + float(j[:-1]) * 0.0001
           else:
               actor_totalnum[i] = actor_totalnum.get(i, 0.0) + float(j[:-1])
       actor_totalnum_temp = [(k, actor_totalnum[k]) for k in actor_totalnum]
       actor_totalnum_temp.sort(key=lambda k: k[1], reverse=True)
       print(actor_totalnum_temp)
       for actor, totalnum in actor_totalnum_temp:
           writer.writerow((actor, totalnum))
複製代碼

主函數

def main():
   ''' 主函數 '''
   #獲取2018內地票房（國產片）的電影列表
   movies = getMovieList()
   #將獲取的2018內地票房（國產片）的電影列表寫入csv文件中
   #write_movie_dsv(movies)
   #統計2018內地（國產片）演員的票房，並寫入csv文件中
   write_actorTotalNum_dsv(movies)
   #統計top10電影
   #movie_data = pd.read_csv('movie_2018_chinalist.csv')
   #統計2018內地(國產片)演員票房排行榜top10
   actor_data=pd.read_csv('actor_totalnum_2018.csv')
   # top10
   top10_actor = actor_data.head(10)
   # Kind表示畫什麼類型圖（柱狀圖等）
   top10_actor.plot(kind='bar', x='演員', y='票房(億)', title='2018內地(國產片)演員票房排行榜top10', figsize=(20, 10))
   plt.savefig('top10_actor2018_bar.png')
   plt.show()
   # 保存爲csv文件
   top10_actor.to_csv('top10_actor2018.csv', index=False)


if __name__=='__main__':
   main()
複製代碼

若是想要統計2019票房王，能夠直接修改Params參數year爲2019，可是下方的Page頁數要根據實際狀況修改成對應的頁數。