实战|数据分析篇之豆瓣电影 TOP250

上次我们对豆瓣 TOP250 电影进行了抓取,链接我放在文末,需要自取。今天我们就对这批数据分析一波,看看可以找到什么结论。

今天主要分析以下几个点。

什么类型的电影上榜数量最多。

上榜数量最多的国家和地区是哪里。

上榜次数最多的导演和演员都有谁。

电影的排名和评论人数以及评分人数有没有关系。

上榜电影中人们更喜欢用哪些标签给电影做标注。

数据清洗

一般来说我们得到的数据都不是可以直接拿来现用的,因为里面可能存在着空值,重复值,异常值等各种情况。这些统称为脏数据,所以我们第一步就要对脏数据做清洗,将其转化为合格数据。

我们获取到的数据都是以 json 串的格式存放在一个 txt 文件中。先将这些数据读取出来,放入到 DataFrame 中去。

数据格式如下

{'index': 1, 'title': '肖申克的救赎 The Shawshank Redemption', 'url': 'https://movie.douban.com/subject/1292052/', 'director': '弗兰克·德拉邦特', 'actor': '蒂姆·罗宾斯#摩根·弗里曼#鲍勃·冈顿#威廉姆·赛德勒#克兰西·布朗#吉尔·贝罗斯#马克·罗斯顿#詹姆斯·惠特摩#杰弗里·德曼#拉里·布兰登伯格#尼尔·吉恩托利#布赖恩·利比#大卫·普罗瓦尔#约瑟夫·劳格诺#祖德·塞克利拉#保罗·麦克兰尼#芮妮·布莱恩#阿方索·弗里曼#V·J·福斯特#弗兰克·梅德拉诺#马克·迈尔斯#尼尔·萨默斯#耐德·巴拉米#布赖恩·戴拉特#唐·麦克马纳斯', 'country': '美国', 'year': '1994', 'type': '剧情#犯罪', 'comments': '全部 340688 条', 'runtime': '142分钟', 'average': '9.7', 'votes': '1885235', 'rating_per': '85.0%#13.4%', 'tags': '经典#励志#信念#自由#人性#人生#美国#希望'}

首先导入我们今天需要用到的包。

import numpy as npimport pandas as pdimport matplotlib.pyplot as pltimport matplotlibfrom wordcloud import WordCloud
content = []
with open(file) as f: line = f.readline() while line: line = eval(line) content.append(line) line = f.readline()
d = pd.DataFrame(content)

下面来看看数据的基本信息。

print(d.info)print(len(d.title.unique()))
# 结果如下<class 'pandas.core.frame.DataFrame'>RangeIndex: 250 entries, 0 to 249Data columns (total 14 columns):actor 250 non-null objectaverage 250 non-null objectcomments 250 non-null objectcountry 250 non-null objectdirector 250 non-null objectindex 250 non-null int64rating_per 250 non-null objectruntime 250 non-null objecttags 250 non-null objecttitle 250 non-null objecttype 250 non-null objecturl 250 non-null objectvotes 250 non-null objectyear 250 non-null objectdtypes: int64(1), object(13)memory usage: 27.4+ KBNone250

共计 250 行,14 列,除 index 为 int 类型之外,其余全是 object 类型,没有缺失值,且没有重复名字的电影,说明数据是完整的。

咱们先来看康什么类型的电影上榜数量最多。

因为一个电影往往有较多的类型标签,所以我们需要对数据做一下分割。

types = d['type'].str.split('#', expand=True)print(types)
# 输出结果 0 1 2 3 40 剧情 犯罪 None None None1 剧情 爱情 同性 None None......248 剧情 None None None None249 动作 科幻 惊悚 犯罪 None

进过分割操作之后我们发现,有的电影多达五个标签,对于这么多的 None 值,可以先按列计数,然后将空值 None 替换为 0,最后再按行汇总,统计出每个类型的总数即可。

types.columns = ['zero', 'one', 'two', 'three', 'four']# 按列计数,并填充 0types = types.apply(pd.value_counts).fillna(0)# 按行计数,统计汇总types['counts'] = types.apply(lambda x: x.sum(), axis=1)# 排序types = types.sort_values('counts', ascending=False)print(types.head(10))
# 输出结果 zero one two three four counts剧情 186.0 0.0 0.0 0.0 0.0 186.0爱情 1.0 42.0 12.0 0.0 0.0 55.0喜剧 21.0 30.0 0.0 0.0 0.0 51.0犯罪 0.0 17.0 19.0 8.0 2.0 46.0冒险 0.0 2.0 30.0 10.0 2.0 44.0奇幻 2.0 16.0 16.0 5.0 0.0 39.0惊悚 0.0 11.0 18.0 6.0 0.0 35.0动画 13.0 14.0 5.0 2.0 0.0 34.0动作 14.0 15.0 3.0 0.0 0.0 32.0悬疑 3.0 24.0 4.0 0.0 0.0 31.0

剧情,爱情,喜剧占据榜首。大多数男孩子喜欢的动作电影上榜数量并不多。

同样的操作,我们对国家地区分析下,看看哪个国家上榜数量最多。

d['country'] = d['country'].str.replace(' ', '')country = d['country'].str.split('/', expand=True)
country.columns = ['zero', 'one', 'two', 'three', 'four', 'five']country = country.apply(pd.value_counts).fillna(0)country['counts'] = country.apply(lambda x: x.sum(), axis=1)country = country.sort_values('counts', ascending=False)print(country.head(10))
# 输出结果 zero one two three four five counts美国 118.0 13.0 3.0 4.0 0.0 0.0 138.0日本 32.0 2.0 0.0 0.0 0.0 0.0 34.0英国 14.0 15.0 4.0 0.0 0.0 0.0 33.0中国香港 18.0 8.0 0.0 1.0 0.0 0.0 27.0中国大陆 16.0 5.0 1.0 0.0 0.0 0.0 22.0法国 8.0 10.0 1.0 1.0 0.0 0.0 20.0德国 5.0 10.0 3.0 0.0 0.0 1.0 19.0韩国 10.0 0.0 1.0 0.0 0.0 0.0 11.0意大利 6.0 2.0 1.0 0.0 0.0 0.0 9.0中国台湾 6.0 2.0 0.0 0.0 0.0 0.0 8.0

美国以 138 个高居榜首,不错的是中国大陆,中国香港和中国台湾都在 TOP10。其中中国大陆排第五。

下面我们看下是哪位天才导演的作品上榜数量最多。

虽说导演数据也是需要分割的,完全可以按照上面两个例子照葫芦画瓢,但这次我们换个方式来。

# 分割数据directors = d['director'].str.split('#').apply(pd.Series)# 行列转换,并重置 indexdirectors = directors.unstack().dropna().reset_index()directors.columns.values[2] = 'name'# 统计导演作品数量directors = directors.name.value_counts()
print(directors.head(10))
# 输出结果宫崎骏 7史蒂文·斯皮尔伯格 7克里斯托弗·诺兰 7王家卫 5李安 5大卫·芬奇 4是枝裕和 4彼得·杰克逊 3朱塞佩·托纳多雷 3弗朗西斯·福特·科波拉 3Name: name, dtype: int64

其中宫崎骏,斯皮尔伯格以及诺兰以 7 部作品并列第一,王家卫和李安以 5 部作品并列第二。

最后我们看看演员的上榜数据如何。

actor = d['actor'].str.split('#').apply(pd.Series)

由于演员数量巨大,所以我们只分析前三列。

actor = d['actor'].str.split('#').apply(pd.Series)[[0, 1, 2]]actor = actor.unstack().dropna().reset_index()actor.columns.values[2] = 'name'actor = actor.name.value_counts()
print(actor.head(10))
# 输出结果张国荣 8梁朝伟 7汤姆·汉克斯 6莱昂纳多·迪卡普里奥 6布拉德·皮特 5周星驰 5张曼玉 5伊桑·霍克 5林青霞 4马特·达蒙 4Name: name, dtype: int64

上榜次数最多的是张国荣哥哥,高达 8 次,一个人演绎了这么多经典作品,不愧是我们永远的哥哥。第二是梁朝伟。第一第二都是咱中国的演员,骄傲了。

数据分析

分别按照评分人数和评论人数取 TOP10 的电影数据来看看。

按照评分人数排序

d['votes'] = d['votes'].astype(int)top10_votes_movie = d[['title', 'votes']].sort_values('votes', ascending=False).head(10).reset_index()print(top10_votes_movie)
index title votes0 0 肖申克的救赎 The Shawshank Redemption 18852351 3 这个杀手不太冷 Léon 16321402 6 千与千寻 千と千尋の神隠し 14732963 2 阿甘正传 Forrest Gump 14369464 53 我不是药神 14003975 8 盗梦空间 Inception 13875166 1 霸王别姬 13843037 5 泰坦尼克号 Titanic 13800738 12 三傻大闹宝莱坞 3 Idiots 12732509 18 疯狂动物城 Zootopia 1182866

按照评论人数排序

d['comments'] = d['comments'].str.split(' ').apply(pd.Series)[1]d['comments'] = d['comments'].astype(int)
top10_comments_movie = d[['title', 'comments']].sort_values('comments', ascending=False).head(10).reset_index()print(top10_comments_movie)
# 输出结果 index title comments0 53 我不是药神 3886541 0 肖申克的救赎 The Shawshank Redemption 3406882 1 霸王别姬 2744903 3 这个杀手不太冷 Léon 2685914 23 怦然心动 Flipped 2636145 85 绿皮书 Green Book 2576106 8 盗梦空间 Inception 2573057 32 寻梦环游记 Coco 2532928 6 千与千寻 千と千尋の神隠し 2518099 166 头号玩家 Ready Player One 248538

可以看出,在评分人数和评论人数方面「肖申克的救赎」都很稳,榜单排名第二的「霸王别姬」在评分人数和评论人数的排名上分别是第八和第三,有点惊讶。

比较惊讶的是榜单排名第 54 位的「我不是药神」,其评论人数和评分人数都相当多,尤其是评论人数,已经超过了很久之前上映的「肖申克的救赎」,而「我不是药神」则是在 2018 年刚上映的。

排名与评分人数的关系

plt.figure(figsize=(20,5))plt.subplot(1,2,1)
# 绘制散点图plt.scatter(d['votes'],d['index'])plt.xlabel('votes')plt.ylabel('rank')plt.gca().invert_yaxis()
# 绘制直方图plt.subplot(1,2,2)plt.hist(d['votes'])

从上图可以看出,评分人数大都集中在 250000 左右,二者呈现强相关性,相关系数为 -0.655。

排名与评论人数的关系

plt.figure(figsize=(20,5))plt.subplot(1,2,1)
# 绘制散点图plt.scatter(d['comments'],d['index'])plt.xlabel('comments')plt.ylabel('rank')plt.gca().invert_yaxis()
# 绘制直方图plt.subplot(1,2,2)plt.hist(d['comments'])

从上图可以看出,评论人数大都集中在 40000~120000 左右,二者相关系数为 -0.539。

类型

最招人喜欢的类型是剧情,其次是爱情,看来爱情是人类永恒的需求啊。

# 设置字体,不然中文会乱码my_font = font_manager.FontProperties(fname='/System/Library/Fonts/PingFang.ttc')plt.figure(figsize=(20,6))plt.title("类型&电影数量", fontproperties=my_font)plt.xticks(fontproperties=my_font,rotation=45)plt.bar(types.index.values, types['counts'])

国家和地区

美国数量最多,有压倒性优势,中国香港第四,中国大陆第五。

plt.figure(figsize=(20,6))plt.title("国家&电影数量", fontproperties=my_font)plt.xticks(fontproperties=my_font,rotation=45)plt.bar(country.index.values, country['counts'])

标签

最后,因为标签数量太大,所以我们可用 WordCloud 将标签制作一个词云图。

tags = d['tags'].str.split('#').apply(pd.Series)text = tags.to_string(header=False,index=False)
wc = WordCloud(font_path = '/System/Library/Fonts/PingFang.ttc',background_color="white",scale=2.5,contour_color="lightblue",).generate(text)wordcloud = WordCloud(background_color='white',scale=1.5).generate(text)plt.figure(figsize=(16,9))plt.imshow(wc)plt.axis('off')plt.show()

总结

今天我们用 pandasmatplotlib 以及 wordcloud 三个库对豆瓣 TOP250 电影数据进行了一波分析,难点主要就是数据的清洗了,把格式错误的数据转化成我们需要的格式,其次就是 DataFrame 和 Series 的操作。

由以上分析我们可以得出,豆瓣电影 TOP250 排行榜和电影评分及评论人数有较强的相关性,美国的电影上榜数量最多。

上榜次数最多的主演是张国荣,上榜次数最多的导演是宫崎骏,斯皮尔伯格以及诺兰。

剧情、爱情类的电影最受欢迎。

代码地址

示例代码:https://github.com/JustDoPython/python-100-day/tree/master/douban-movie-top250

第119天:Python 爬取豆瓣电影 top 250

(0)

相关推荐